mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-09-12 12:40:20 +08:00
updated args
This commit is contained in:
@@ -1,5 +1,6 @@
|
||||
from itertools import chain
|
||||
from torch.nn import DataParallel
|
||||
import pdb
|
||||
|
||||
|
||||
class LightningDataParallel(DataParallel):
|
||||
@@ -8,6 +9,7 @@ class LightningDataParallel(DataParallel):
|
||||
"""
|
||||
|
||||
def forward(self, *inputs, **kwargs):
|
||||
pdb.set_trace()
|
||||
if not self.device_ids:
|
||||
# -------------
|
||||
# MAIN CHANGE
|
||||
|
||||
Reference in New Issue
Block a user