federated_main.py not working

[2ch42]

Hi I tried to run "python src/federated_main.py --model=cnn --dataset=cifar --gpu=0 --iid=1 --epochs=10"
but is not working. (in any option w federate_main.py including dataset, model, so)

I found several issues from your git and modified those parts, but it seems like there r additional problem w loop of 'federated_main.py'.

Is there anyone else who r suffering from same issue or have fixed them?

[waaki7]

Hi, I have the same issue with non-iid dataset, CIFAR10 object has no attribute 'train_labels'. Did you manage to solve it @2ch42

[nguyenhongson1902]

Hi, I have the same issue with non-iid dataset, CIFAR10 object has no attribute 'train_labels'. Did you manage to solve it @2ch42

Because dataset.train_labels is deprecated in the latest version of PyTorch, that's why you saw this error. So to deal with it, instead of using train_labels, you should use dataset.targets to get the labels. That worked for me, hope it works for you as well.