Skip to content
Snippets Groups Projects
Commit 4c354428 authored by Jeffrey Wigger's avatar Jeffrey Wigger
Browse files

running with reddit 6

parent 69353274
No related branches found
No related tags found
No related merge requests found
......@@ -24,11 +24,14 @@ loss_package = torch.nn
loss_class = CrossEntropyLoss
[COMMUNICATION]
comm_package = decentralizepy.communication.TCPRandomWalkRouting
comm_class = TCPRandomWalkRouting
comm_package = decentralizepy.communication.TCPRandomWalk
comm_class = TCPRandomWalk
addresses_filepath = ip_addr_6Machines.json
sampler = equi
[SHARING]
sharing_package = decentralizepy.sharing.SharingDynamicGraph
sharing_class = SharingDynamicGraph
\ No newline at end of file
sharing_package = decentralizepy.sharing.TopKDPSGDRW
sharing_class = TopKDPSGDRW
alpha = 1.0
accumulation = True
accumulate_averaging_changes = True
\ No newline at end of file
......@@ -512,7 +512,7 @@ class Node:
logging.info("disconnect neighbors")
self.communication.disconnect_neighbors()
logging.info("Storing final weight")
self.model.dump_weights(self.weights_store_dir, self.uid, iteration)
# self.model.dump_weights(self.weights_store_dir, self.uid, iteration)
logging.info("All neighbors disconnected. Process complete!")
def __init__(
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment