diff --git a/src/node_manager.py b/src/node_manager.py index b6f8cc6..1d0d5c9 100644 --- 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)