snsd0805/gpu-provider
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: uncommand and let the provider listen events

Browse Source
This commit is contained in:
Ting-Jun Wang 2024-06-02 23:17:57 +08:00
parent d7504af42d
commit 361988c079
Signed by: snsd0805
GPG Key ID: 48D331A3D6160354
1 changed files with 4 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
src/node_manager.py
View File

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