Feature_Critic icon indicating copy to clipboard operation
Feature_Critic copied to clipboard

Other networks

Open Epi-NMT opened this issue 5 years ago • 3 comments

Can this code apply to other pretrained networks? like resnet18, resnet50. Based on the line 361 in model_PACS, it seems it cannot apply on resnet18. I have tried resnet50, but there are many details need to be modified.

Epi-NMT avatar Oct 31 '19 21:10 Epi-NMT

Yes, it can be applied to other networks. Line 361 here is not to emphasize the version of Resnet "18". The batchnorm function is used in the BasicBlock/Bottleneck which are used by different versions of ResNet. If you read the source code of ResNet carefully, you won't have such questions. Well, actually you need to modify some details if you use other networks. I believe you can find these places to modify by debugging.

liyiying avatar Nov 01 '19 15:11 liyiying

Have you ever try your method on resnet18 or resnet50? How is the performance?

Epi-NMT avatar Nov 03 '19 17:11 Epi-NMT

I have tried some different versions of ResNet on VD instead of PACS, and see some slight improvement with more layers of ResNet.

在 2019年11月3日,下午5:41,keyu07 [email protected] 写道:

Have you ever try your method on resnet18 or resnet50? How is the performance?

— You are receiving this because you commented. Reply to this email directly, view it on GitHub, or unsubscribe.

liyiying avatar Nov 04 '19 00:11 liyiying