search

crywang / RFM

Apache License 2.0
83 stars 11 forks source link

issue of Xception #3

Closed huiyegit closed 2 years ago

huiyegit commented 2 years ago

Thank you for sharing your source code. I tried running the source code and had an issue. It seems there is something wrong with the line '' model = eval(modelname)(num_classes=2).cuda()'' . Is the definition of the Xception correct in the source code? Selection_019

crywang commented 2 years ago

You can try changing the code to eval(modelname)(num_classes=2, pretrained=False).cuda().

huiyegit commented 2 years ago

Thank you for your answer.