Train fails on each and every command

ssnl / dataset-distillation

Open-source code for paper "Dataset Distillation"

https://ssnl.github.io/dataset_distillation

MIT License

778 stars 115 forks source link

Train fails on each and every command #62

Closed gonenraveh closed 2 months ago

gonenraveh commented 2 months ago

I ran: python main.py --mode distill_basic --dataset Cifar10 --arch AlexCifarNet --distill_lr 0.001 --train_nets_type known_init --n_nets 1 --test_nets_type same_as_trai

and got: File "cifar_100_dataset_distillation/dataset-distillation/train_distilled_image.py", line 176, in accumulategrad d.grad.add(g) AttributeError: 'NoneType' object has no attribute 'add_'

Please advise. Gonen

ssnl commented 2 months ago

We haven't tested this code on later pytorch versions. If you see problems, please try using pytorch 1.0.0.