From 471ca375babad9093abf60683a8d0647ac33d4a8 Mon Sep 17 00:00:00 2001 From: Jungwhan <53588015+jungwhank@users.noreply.github.com> Date: Tue, 24 Nov 2020 14:27:14 +0900 Subject: [PATCH] Fix torchtext data to gpu (#4785) Co-authored-by: chaton --- pytorch_lightning/utilities/apply_func.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/pytorch_lightning/utilities/apply_func.py b/pytorch_lightning/utilities/apply_func.py index 4f2212ba2b..775c22dbbf 100644 --- a/pytorch_lightning/utilities/apply_func.py +++ b/pytorch_lightning/utilities/apply_func.py @@ -113,7 +113,9 @@ def move_data_to_device(batch: Any, device: torch.device): # Shallow copy because each Batch has a reference to Dataset which contains all examples device_data = copy(data) - for field in data.fields: + for field, field_value in data.dataset.fields.items(): + if field_value is None: + continue device_field = move_data_to_device(getattr(data, field), device) setattr(device_data, field, device_field) return device_data