Hi, I am trying to compare for a paper and I couldn't get it to work on resnet18, can you please share your resnet18 code so I can run your model?