adding support for interrupt signals

This commit is contained in:
William Falcon 2019-06-14 09:28:23 -04:00
parent 8cca02d652
commit 519d2e9321
1 changed files with 1 additions and 1 deletions

View File

@ -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