MichalisLazarou / iLPC

accepted at ICCV 2021
MIT License
20 stars 2 forks source link

It is TOO SLOW because of the label_denoising function #5

Closed machengcheng2016 closed 1 year ago

machengcheng2016 commented 1 year ago

Greetings, I've been trying to reproduce the reported results in your paper. However, when I run python semi_resnet12.py --dataset miniImagenet --algorithm ilpc --alpha 0.8 --K 15 --n_shots 1, I find that the label_denoising function cost ~50 minutes on a single episode, which means >800 hours on 1000 episodes to obtain the final results. The reason lies in that there are 1000 iterations within the label_denoising function. I wonder if you ever face the same problems? Is there any method to boost the computation? Thanks

machengcheng2016 commented 1 year ago

Never mind. I put the label_denoising function on GPU and now it is much faster (about 4.5 second per episode).