Hyper Parameter for Resnet18 Model

[ganguli9082]

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