adding support for interrupt signals

2026-09-26 14:11:01 +08:00 · 2019-06-14 09:28:23 -04:00
parent 8cca02d652
commit 519d2e9321
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 = [0, 1, 2, 3, 4, 5, 6, 7]
+        self.data_parallel_device_ids = []

        # training state
        self.optimizers = None