AttributeError: module 'torch' has no attribute 'amp'

alanqrwang / keymorph

Robust multimodal image registration via keypoints

MIT License

67 stars 17 forks source link

Closed lxy51 closed 6 months ago

lxy51 commented 10 months ago

AttributeError: module 'torch' has no attribute 'amp' How should this problem be solved

alanqrwang commented 6 months ago

Sorry for the delayed response. torch.amp requires at least Pytorch version 1.10. Closing now, feel free to open another issue if necessary.