search

yaoyao-liu / meta-transfer-learning

TensorFlow and PyTorch implementation of "Meta-Transfer Learning for Few-Shot Learning" (CVPR2019)
https://lyy.mpi-inf.mpg.de/mtl/
MIT License
731 stars 147 forks source link

Performance Pytorch Version #46

Closed mikehuisman closed 3 years ago

mikehuisman commented 3 years ago

Hi Yaoyao,

In your repository, it says that the pytorch performance is under check. Do you have any update on this?

Kind regards,

Mike Huisman

yaoyao-liu commented 3 years ago

Thanks for your interest in our work.

Could you please try the pre-train model/code in this project? We're using a similar pre-training strategy. So that we can find which part is incorrect.

yaoyao-liu commented 3 years ago

I have tested the current code. The performance is around 59%-61%. It shouldn't be that low. Related issue: link

mikehuisman commented 3 years ago

Thank you for your prompt response! Yesterday, I reinstalled the environment, re-downloaded all data sets, updated CUDA drivers, and now I indeed get the same results! Thank you for your help.