moved cuda flags inside trainer

2026-09-11 12:31:23 +08:00 · 2019-07-08 09:58:43 -04:00
parent 2172280d8d
commit 79ca5f6265
1 changed files with 0 additions and 1 deletions
@@ -90,7 +90,6 @@ class Trainer(TrainerIO):
        # gpus come in as a string.
        # if gpus = -1 then use all available devices
        # otherwise, split the string using commas
-        pdb.set_trace()
        if gpus is not None:
            if gpus == '-1':
                self.data_parallel_device_ids = list(range(0, torch.cuda.device_count()))