We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 4658299 commit 684d152Copy full SHA for 684d152
pytorch_lightning/utilities/apply_func.py
@@ -113,7 +113,9 @@ def batch_to(data):
113
114
# Shallow copy because each Batch has a reference to Dataset which contains all examples
115
device_data = copy(data)
116
- for field in data.fields:
+ for field, field_value in data.dataset.fields.items():
117
+ if field_value is None:
118
+ continue
119
device_field = move_data_to_device(getattr(data, field), device)
120
setattr(device_data, field, device_field)
121
return device_data
0 commit comments