working on single gpu init speed

2026-09-11 12:31:23 +08:00 · 2019-07-14 17:11:52 -04:00
parent 468e75c180
commit 904935cf98
1 changed files with 1 additions and 0 deletions
@@ -306,6 +306,7 @@ class Trainer(TrainerIO):
            self.single_gpu_train(model)

        else:
+            print('NO GPU')
            # CHOOSE OPTIMIZER
            # filter out the weights that were done on gpu so we can load on good old cpus
            self.optimizers = model.configure_optimizers()