'collections. OrderedDict' object has no attribute 'to'.

[hitzhangyu]

HI，
The work is amazing, but I have a problem using the pretrain_models：'collections. OrderedDict' object has no attribute 'to'.
It seems that only the "model.state_dict()" is provided. Did the author use the “torch.save(model.state_dict(),model_path)”
instead of "torch.save(model,' save_path')" when saving the model?

[amberwangyili]

Thanks very much for your attention!
"Did the author use the “torch.save(model.state_dict(),model_path)” instead of "torch.save(model,' save_path')" when saving the model?" ---- Yes, when saving the model, only model.state_dict() is provided.
The released pretrained weights is not the official version. The official code & model checkpoint with configurations will be released soon.

[hitzhangyu]

Thanks for your reply!