added option to change default tensor

2026-09-10 12:21:57 +08:00 · 2019-05-13 21:19:07 -04:00
parent 5f0a71c414
commit 8e49fc6cf7
1 changed files with 1 additions and 0 deletions
@@ -219,6 +219,7 @@ class Trainer(TrainerIO):
        # filter out the weights that were done on gpu so we can load on good old cpus
        self.optimizers = model.configure_optimizers()

+        pdb.set_trace()
        if self.use_amp:
            # An example
            self.model, optimizer = amp.initialize(