set dp as default backend

This commit is contained in:
William Falcon
2019-07-18 11:25:02 -04:00
parent ded0abead7
commit 551daca047
2 changed files with 1 additions and 1 deletions
-1
View File
@@ -255,7 +255,6 @@ class Trainer(TrainerIO):
# -----------------
# RUN VALIDATION STEP
# -----------------
pdb.set_trace()
if self.data_parallel:
output = model(data_batch, batch_i)
else:
@@ -130,6 +130,7 @@ def parallel_apply(modules, inputs, kwargs_tup=None, devices=None):
# ---------------
# CHANGE
pdb.set_trace()
if module.training:
output = module.training_step(*input, **kwargs)
else: