adding support for interrupt signals
This commit is contained in:
parent
cd36b63167
commit
32edc6d7b7
|
@ -63,8 +63,8 @@ class Trainer(TrainerIO):
|
|||
self.lr_schedulers = []
|
||||
self.amp_level = amp_level
|
||||
self.check_grad_nans = check_grad_nans
|
||||
self.data_parallel = True
|
||||
self.data_parallel_device_ids = [0]
|
||||
self.data_parallel = False
|
||||
|
||||
# training state
|
||||
self.optimizers = None
|
||||
|
|
Loading…
Reference in New Issue