Hi, I am using Catalyst for training my model. It works great if I explicitly set CUDA_VISIBLE_DEVICES
, however if I did not set it, I would have following error:
ModuleAttributeError: 'DataParallel' object has no attribute '_loss'
Looks like that the model has been automatically converted to DataParallel
object in Catalyst runner while the corresponding _loss
method of my model was not assigned to it. Does anyone know if there is a way to either:
- correctly assign
_loss
method of the model toDataParallel
object or - disable conversion to
DataParallel
object in Catalyst runner?
Thank you!