Add saveimg

chizuchizu · Mar 25, 2021 · 28a1ae7 · 28a1ae7
1 parent 08649ae
commit 28a1ae7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/mnist.py b/src/mnist.py
@@ -65,7 +65,7 @@ def get_ds(n, seed):
 
 
 dataset = pd.read_csv(cfg.base.train_path)  # .sample(100).values
-for i in range(6, 10):
+for i in range(10):
     init_client(cfg)
     cfg.dataset.target = i
     train = get_ds(