diff --git a/src/node_manager.py b/src/node_manager.py index 460b158..141bf19 100644 --- a/src/node_manager.py +++ b/src/node_manager.py @@ -85,7 +85,6 @@ class NodeManager(): print(f"And we have load your wallet private key {WALLET_KEY} (address={self.wallet.address})") print() if self.w3.is_connected(): - ''' print("[INFO] Connected Successfully.") print() @@ -115,15 +114,15 @@ class NodeManager(): train_image = next_task['args']['trainImage'] print("\n[INFO] You Receive a new task:") + print(f" - Task Index: {task_index}") print(f" - Download Image: {data_image}") print(f" - Training Image: {train_image}") - ''' - data_image = "snsd0805/cifar100-dataset:v1" - train_image = "snsd0805/cifar100-train:v3" + # data_image = "snsd0805/cifar100-dataset:v1" + # train_image = "snsd0805/cifar100-train:v3" # Start Downloading - # self.cluster_communication_module.scatter_container(data_image, train=False) + self.cluster_communication_module.scatter_container(data_image, train=False) # start training self.cluster_communication_module.scatter_container(train_image, train=True)