JWFanggit / LOTVS-CAP

23 stars 4 forks source link

Problems when run test.py #8

Closed syf66888 closed 6 days ago

syf66888 commented 2 weeks ago

You have done a great job, congratulations. But it seems that there is a problem with the test.py you provided. After I changed the downsampling to upsampling, and use model.load_state_dict(weight,strict=False) loaded the pre-trained model you provided and ran test.py. The print "pred" is {'total_loss': tensor(287.2326 , device='cuda:0')} and subsequent testing cannot be performed.

syf66888 commented 2 weeks ago

the problem above i have solved but i have another problem when running test.py, the auc cannot be calculated as below 屏幕截图 2024-10-17 181306

syf66888 commented 6 days ago

I fixed the issue and it seems that I was using the text message incorrectly, so I closed this issue