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
_lossmethod of the model to
- disable conversion to
DataParallelobject in Catalyst runner?