mi-prometheus icon indicating copy to clipboard operation
mi-prometheus copied to clipboard

Published 20 hours ago • verified publisher icon IBM

Pin_memory: True in config causes error.

Open sesevgen opened this issue 6 years ago • 1 comments

Describe the bug Setting pin_memory: True in yaml under dataloader causes an AttributeError: "AttributeError: 'dict' object has no attribute 'cuda'"

To Reproduce Steps to reproduce the behavior: Set in a configuration file: training: dataloader: pin_memory:True

and run with GPU enabled.

Desktop (please complete the following information):

  • OS: Ubuntu 16.04.5 LTS (Newton)
  • Python version 3.6
  • PyTorch version 4.1

sesevgen avatar Jan 07 '19 23:01 sesevgen