added training router

2026-09-10 12:21:57 +08:00 · 2019-07-18 11:09:00 -04:00
parent 470f3e6d29
commit baa139f97a
1 changed files with 1 additions and 0 deletions
@@ -319,6 +319,7 @@ class Trainer(TrainerIO):
        elif self.use_dp:
            self.dp_train(model)

+        # ON CPU
        else:
            # CHOOSE OPTIMIZER
            # filter out the weights that were done on gpu so we can load on good old cpus