fixed devices

kalininalab · Aug 16, 2024 · 6b10de7 · 6b10de7
1 parent bf48266
commit 6b10de7
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/smtb/train.py b/smtb/train.py
@@ -22,7 +22,6 @@ def train(config: argparse.Namespace):
 
     # define the Trainer and it's most important arguments
     trainer = Trainer(
-        devices=-1,
         max_epochs=config.max_epoch,
         callbacks=callbacks,
         logger=logger,