Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

How to use the pretrained model? #42

Open
yjsb opened this issue May 13, 2022 · 2 comments
Open

How to use the pretrained model? #42

yjsb opened this issue May 13, 2022 · 2 comments

Comments

@yjsb
Copy link

yjsb commented May 13, 2022

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
Copy link
Author

yjsb commented May 13, 2022

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

@yjsb
Copy link
Author

yjsb commented May 13, 2022

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

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant