NoduleNet icon indicating copy to clipboard operation
NoduleNet copied to clipboard

How to use the pretrained model?

Open yjsb opened this issue 3 years ago • 2 comments

I downloaded the file '200.ckpt' on Google Drive, and I changed its name to '001.ckpt' in my project. I trained the model for 50 epoches on fold 3, but I get a poor mean dice(exclude fn); 0.74, which in the paper is 0.83. So it maybe something wrong in my training procedure, can u help me, pls?

yjsb avatar May 13 '22 08:05 yjsb

The learning rate I used in my training procedure was 0.0001, I think it is small enough.

yjsb avatar May 13 '22 08:05 yjsb

I also tested it on 0_val.csv, and got the mean dice(exclude fn): 0.742. The model is download file '200.ckpt'.

yjsb avatar May 13 '22 08:05 yjsb