Updated pre-trained model

Hi thanks for sharing your work, I noticed latest commits use different dataset, any chance you can upload the updated pretrained model?

Current I tested was https://github.com/ShawnNew/AMMNet/blob/master/pretrained.pth