Pass args into trainer properly

fairy-stockfish · Nov 15, 2020 · 483491f · 483491f
1 parent 01c682c
commit 483491f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/train.py b/train.py
@@ -54,7 +54,7 @@ def main():
     train, val = data_loader_cc(args.train, args.val)
 
   tb_logger = pl_loggers.TensorBoardLogger('logs/')
-  trainer = pl.Trainer(logger=tb_logger, gpus=args.gpus)
+  trainer = pl.Trainer.from_argparse_args(args, logger=tb_logger)
   trainer.fit(nnue, train, val)
 
 if __name__ == '__main__':