Kaynağa Gözat

Delete averaging.py

Ashwin R Jadhav 4 yıl önce
ebeveyn
işleme
486bdb43a4
1 değiştirilmiş dosya ile 0 ekleme ve 18 silme
  1. 0 18
      src/averaging.py

+ 0 - 18
src/averaging.py

@@ -1,18 +0,0 @@
-#!/usr/bin/env python
-# -*- coding: utf-8 -*-
-# Python version: 3.6
-
-import copy
-import torch
-
-
-def average_weights(w):
-    """
-    Returns the average of the weights.
-    """
-    w_avg = copy.deepcopy(w[0])
-    for key in w_avg.keys():
-        for i in range(1, len(w)):
-            w_avg[key] += w[i][key]
-        w_avg[key] = torch.div(w_avg[key], len(w))
-    return w_avg