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

Hyper Parameter for Resnet18 Model #1

Open
ganguli9082 opened this issue Dec 20, 2021 · 2 comments
Open

Hyper Parameter for Resnet18 Model #1

ganguli9082 opened this issue Dec 20, 2021 · 2 comments

Comments

@ganguli9082
Copy link

Hi,

Just curious about the parameters that are used for training the model for polyp type on the 7000um(subsample 224) patches and the grade classification that was done on the 800um patches for both the subsample of 224x224 and no subsample. I started out using the default settings in the train.py but can't seem to get close to the results described in the paper.

Thank you for your time.
Alex Ganguli

Preprocess: HE
Apply Transformations: Training images= True,
Testing images = False

learning rate: 0.01
batch size: 256
num workers: 8
decay factor: 0.1
step size: 20

@carloalbertobarbano
Copy link
Member

carloalbertobarbano commented Dec 22, 2021

Hello, sorry for the delay in the response.

So, the are two main tasks:

  1. Type classification on 7000µm patches -> here we subsample to 224
  2. Grade prediction on 800µm patches -> here we do not subsample the input image

In any case the preprocessing is RGB. Can you try replicating task 2 with one of these hyperparameters settings: https://wandb.ai/eidos/UnitoPath-v1/reports/Grade-predictions--VmlldzoxMzY4NzI5 that should get you around 80% BA for grade prediction (you can click on a single run then go to overview to get the full list of arguments)

@ganguli9082
Copy link
Author

ganguli9082 commented Dec 22, 2021 via email

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

2 participants