adding support for interrupt signals
This commit is contained in:
parent
8cca02d652
commit
519d2e9321
|
@ -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
|
||||
|
|
Loading…
Reference in New Issue