wassname/pytorch-lightning
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-09-09 11:32:07 +08:00
Code Issues Packages Projects Releases Wiki Activity

set dp as default backend

Browse Source
This commit is contained in:
William Falcon
2019-07-18 11:42:47 -04:00
parent f650253cae
commit 3a1525222d
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pytorch_lightning/models/trainer.py
+4 -4
View File
@@ -640,8 +640,8 @@ class Trainer(TrainerIO):
# hook
if self.__is_function_implemented('on_batch_start'):
model = self.__get_model()
response = model.on_batch_start(data_batch)
model_ref = self.__get_model()
response = model_ref.on_batch_start(data_batch)
if response == -1:
return -1
@@ -652,9 +652,9 @@ class Trainer(TrainerIO):
# forward pass
# return a scalar value and a dic with tqdm metrics
if self.use_ddp:
output = model(data_batch, batch_nb)
output = self.model(data_batch, batch_nb)
elif self.use_dp:
output = model(data_batch, batch_nb)
output = self.model(data_batch, batch_nb)
output = reduce_distributed_output(output, len(self.data_parallel_device_ids))
else:
output = self.model.training_step(data_batch, batch_nb)