After one iter training speed become very slow

LiJunnan1992 / MLNT

Meta-Learning based Noise-Tolerant Training

123 stars 29 forks source link

Closed House-Leo closed 4 years ago

House-Leo commented 4 years ago

微信截图_20191107154400 pytorch version = 1.2.0 Tesla v100 CUDA 10.0

I thing your envirment is pytorch<0.4.0, so I have to change code in order to run it. The baseline.py has no problem, but the main.py has.

LiJunnan1992 commented 4 years ago

I'm not sure why it happens, does the speed increase with a smaller num_fast?