diff --git a/smtb/train.py b/smtb/train.py
index 965bf21..1dc7c97 100644
--- 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,