AttributeError: module 'torch.utils' has no attribute '_load_checkpoint_for_ema'

rayleizhu / BiFormer

[CVPR 2023] Official code release of our paper "BiFormer: Vision Transformer with Bi-Level Routing Attention"

https://arxiv.org/abs/2303.08810

MIT License

501 stars 41 forks source link

AttributeError: module 'torch.utils' has no attribute '_load_checkpoint_for_ema' #48

Open lmqomo12 opened 6 months ago

lmqomo12 commented 6 months ago

File "main.py", line 414, in main torch.utils._load_checkpoint_for_ema(model_ema, checkpoint['model_ema']) AttributeError: module 'torch.utils' has no attribute '_load_checkpoint_for_ema' I think it's because the new version of torch has removed this function, what should I do?