kiddyboots216 / CommEfficient

PyTorch for benchmarking communication-efficient distributed SGD optimization algorithms
72 stars 20 forks source link

How to run experiments using the EMNIST dataset? #18

Closed miliable closed 1 day ago

miliable commented 1 week ago

Hello, author. If I want to use the EMNIST dataset for my experiments, do I need to download the data in advance? After downloading the dataset, I ran the following command:python cv_train.py --dataset_name EMNIST --model ResNet101LN --mode fedavg --num_results_train 2 --train_dataloader_workers 8 --val_dataloader_workers 8 --num_devices 4 --error_type none --lr_scale 0.4 --num_workers 10 --local_momentum 0 --num_clients 1000 --local_batch_size -1 However, the accuracy is quite low, and it stops automatically after 5 epochs. Could you provide the correct command to run FEMNIST, or suggest modifications to improve it? image