AttributeError: 'DIMModel' object has no attribute 'module'

[lfxx]

When resume training using checkpoints,get error above raised by line 38 from train.py:
'model = checkpoint['model'].module',any suggestions?@foamliu

[foamliu]

If you're training with just one GPU, you can remove that ".module" part.