fix: dataset typo

2024-06-01 00:53:33 +08:00 · 2024-06-01 00:53:33 +08:00 · c1c946d01b
commit c1c946d01b
parent 3047b898ae
1 changed files with 1 additions and 1 deletions
--- a/src/node_manager.py
+++ b/src/node_manager.py
@ -119,7 +119,7 @@ class NodeManager():
            print(f"      - Training Image: {train_image}")
            '''

-            data_image = "snsd0805/cifar10-dataset:v1"
+            data_image = "snsd0805/cifar100-dataset:v1"

            # Start Downloading
            self.cluster_communication_module.scatter_container(data_image)