-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathex_pill.sh
7 lines (7 loc) · 2.31 KB
/
ex_pill.sh
1
2
3
4
5
6
7
# pill dataset 3
CUDA_VISIBLE_DEVICES=0 python main.py --task pilldataset_3_clustered_N15_K5 --model resnet18 --algorithm fedavg --data_folder "benchmark/pilldataset/data/pill3/pill_dataset" --dataidx_path "benchmark/pilldataset/data/syndata3" --log_folder fedtask --num_rounds 100 --num_epochs 5 --learning_rate 0.001 --proportion 0.3 --batch_size 8 --num_threads_per_gpu 1 --num_gpus 1 --server_gpu_id 0
CUDA_VISIBLE_DEVICES=0 python main.py --task pilldataset_3_clustered_N15_K5 --model resnet18 --algorithm fedprox --data_folder "benchmark/pilldataset/data/pill3/pill_dataset" --dataidx_path "benchmark/pilldataset/data/syndata3" --log_folder fedtask --num_rounds 100 --num_epochs 5 --learning_rate 0.001 --proportion 0.3 --batch_size 8 --num_threads_per_gpu 1 --num_gpus 1 --server_gpu_id 0
CUDA_VISIBLE_DEVICES=0 python main.py --task pilldataset_3_clustered_N15_K5 --model resnet18 --algorithm cadis --data_folder "benchmark/pilldataset/data/pill3/pill_dataset" --dataidx_path "benchmark/pilldataset/data/syndata3" --log_folder fedtask --num_rounds 100 --num_epochs 5 --learning_rate 0.001 --proportion 0.3 --batch_size 8 --num_threads_per_gpu 1 --num_gpus 1 --server_gpu_id 0
CUDA_VISIBLE_DEVICES=0 python main.py --task pilldataset_3_clustered_N15_K5 --model resnet18 --algorithm fedfv --data_folder "benchmark/pilldataset/data/pill3/pill_dataset" --dataidx_path "benchmark/pilldataset/data/syndata3" --log_folder fedtask --num_rounds 100 --num_epochs 5 --learning_rate 0.001 --proportion 0.3 --batch_size 8 --num_threads_per_gpu 1 --num_gpus 1 --server_gpu_id 0
CUDA_VISIBLE_DEVICES=0 python main.py --task pilldataset_3_clustered_N15_K5 --model resnet18 --algorithm scaffold --data_folder "benchmark/pilldataset/data/pill3/pill_dataset" --dataidx_path "benchmark/pilldataset/data/syndata3" --log_folder fedtask --num_rounds 100 --num_epochs 5 --learning_rate 0.001 --proportion 0.3 --batch_size 8 --num_threads_per_gpu 1 --num_gpus 1 --server_gpu_id 0
CUDA_VISIBLE_DEVICES=0 python main.py --task pilldataset_3_clustered_N15_K5 --model resnet18 --algorithm feddyn --data_folder "benchmark/pilldataset/data/pill3/pill_dataset" --dataidx_path "benchmark/pilldataset/data/syndata3" --log_folder fedtask --num_rounds 100 --num_epochs 5 --learning_rate 0.001 --proportion 0.3 --batch_size 8 --num_threads_per_gpu 1 --num_gpus 1 --server_gpu_id 0