updated args

2026-09-11 12:31:23 +08:00 · 2019-06-25 20:28:33 -04:00
parent 5eaaf82837
commit 5ce4e872de
1 changed files with 1 additions and 0 deletions
@@ -201,6 +201,7 @@ class Trainer(TrainerIO):
                output = reduce_distributed_output(output, len(self.data_parallel_device_ids))
            else:
                output = model.validation_step(data_batch, batch_i)
+                pdb.set_trace()
                output = reduce_distributed_output(output, 1)

            outputs.append(output)