Commit e4f866f0 authored by Paul Nadal's avatar Paul Nadal 💻
Browse files

remove tests

parent 7bba867b
import json
import logging
from collections import deque
import time
import numpy
import torch
......@@ -153,8 +152,6 @@ class Sharing:
)
)
start = time.time()
logging.info("Starting model averaging after receiving from all neighbors")
total = dict()
weight_total = 0
......@@ -175,9 +172,6 @@ class Sharing:
for key, value in self.model.state_dict().items():
total[key] += (1 - weight_total) * value # Metro-Hastings
end = time.time()
logging.info("Average execution time : " + str(end - start) + " seconds")
self.model.load_state_dict(total)
logging.info("Model averaging complete")
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment