GeorgeCazenavette / mtt-distillation

Official code for our CVPR '22 paper "Dataset Distillation by Matching Training Trajectories"
https://georgecazenavette.github.io/mtt-distillation/
Other
395 stars 55 forks source link

Update distill.py #19

Closed zhaoguangxiang closed 2 years ago

zhaoguangxiang commented 2 years ago

Solving the bug of "TypeError: 'numpy.float64' object cannot be interpreted as an integer" in Line 111