Hi @charlesnchr, your work is both interesting and inspiring. I am right now trying to run your pretained model, however, I found that the parameter setting used to derive the .pth file seems to be quite different from the setting in the file 3_Evaluate.ipynb, could you please provide the parameter setting used to get the pretrained model? Thanks!