refactored model tests

2026-09-11 12:31:23 +08:00 · 2019-07-24 13:49:28 -04:00
parent 4e6c7f80e5
commit 7d1e1eb7f9
1 changed files with 0 additions and 2 deletions
@@ -458,8 +458,6 @@ 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()

-        model.cuda(self.data_parallel_device_ids[0])
-
        # run through amp wrapper
        if self.use_amp:
            # An example