added cpu example

2026-09-09 11:32:07 +08:00 · 2019-07-08 17:44:06 -04:00
parent 7a354668ff
commit a311a62b48
1 changed files with 1 additions and 0 deletions
@@ -86,6 +86,7 @@ class Trainer(TrainerIO):
        self.lr_schedulers = []
        self.amp_level = amp_level
        self.print_nan_grads = print_nan_grads
+        self.data_parallel_device_ids = None

        # gpus come in as a string.
        # if gpus = -1 then use all available devices