adding support for interrupt signals

2026-09-10 12:21:57 +08:00 · 2019-06-14 09:39:52 -04:00
parent 519d2e9321
commit cd36b63167
1 changed files with 1 additions and 1 deletions
@@ -64,7 +64,7 @@ class Trainer(TrainerIO):
        self.amp_level = amp_level
        self.check_grad_nans = check_grad_nans
        self.data_parallel = True
-        self.data_parallel_device_ids = []
+        self.data_parallel_device_ids = [0]

        # training state
        self.optimizers = None