pytorch-MOAT
pytorch-MOAT copied to clipboard
pytorch实现MOAT(未开源),可以在ImageNet或自己的数据集上训练,支持apex混合精度,各种图像增强技术
moat
请问作者有没有训练好的权重啊,我想把预训练的moat迁移到我的病理图像分类看看效果。如果您愿意分享,万分感谢!
然后呢?效果如何?放个你跑的结果来看看呢?可以跟原论文比较一下 ALTERNATING MOBILE CONVOLUTION AND ATTENTION BRINGS STRONG VISION MODELS
运行错误
torch.distributed.reduce_scatter_tensor = torch.distributed._reduce_scatter_base AttributeError: module 'torch.distributed' has no attribute '_reduce_scatter_base' 运行时提示如上错误 运行的环境: torch 1.9.0 torchvision 0.10.0