diff --git a/README.md b/README.md index 6dd6d36..d9b00bc 100644 --- a/README.md +++ b/README.md @@ -55,6 +55,7 @@ export CUDA_VISIBLE_DEVICES= ``` +mkdir log_synthetic bash run_fedavg.sh synthetic_iid 0 | tee log_synthetic/synthetic_iid_client10_epoch20_mu0 bash run_fedprox.sh synthetic_iid 0 1 | tee log_synthetic/synthetic_iid_client10_epoch20_mu1 bash run_fedavg.sh synthetic_0_0 0 | tee log_synthetic/synthetic_0_0_client10_epoch20_mu0 diff --git a/data/nist/data/my_sample.py b/data/nist/data/my_sample.py index 42b27f9..384ae65 100644 --- a/data/nist/data/my_sample.py +++ b/data/nist/data/my_sample.py @@ -57,7 +57,9 @@ def main(): for class_ in os.listdir(file_dir): - + if class_ == ".DS_Store": + continue + real_class = relabel_class(class_) if real_class >= 36 and real_class <= 45: full_img_path = file_dir + "/" + class_ + "/train_" + class_ diff --git a/data/synthetic_iid/README.md b/data/synthetic_iid/README.md index 536c67d..cfbba39 100644 --- a/data/synthetic_iid/README.md +++ b/data/synthetic_iid/README.md @@ -1,3 +1,3 @@ ``` -python generate_synthetic.py -``` \ No newline at end of file +python generate_iid.py +```