search

xmindflow / deformableLKA

[WACV 2024] Beyond Self-Attention: Deformable Large Kernel Attention for Medical Image Segmentation
https://arxiv.org/abs/2309.00121
157 stars 12 forks source link

can not find best_model.pth and best_model.pth.txt #17

Closed xiexuxiang closed 3 months ago

xiexuxiang commented 3 months ago

Dear author： Due to the poor performance of my computer, it is difficult to finish running the train_MaxViT_deform_LKA.py file. When I run the test.py file it shows that I don't have best_model.pth and best_model.pth.txt Do I need to run train_MaxViT_deform_LKA.py completely to obtain these two files? Or can these two files be downloaded directly?

Leonngm commented 3 months ago

Hi, to run the test file, you need the trained checkpoint. It is provided under this link. You can rename it accordingly. A .txt file should not be necessary. Best Leon