From 5c138837ba671000d6110c265290088141b84774 Mon Sep 17 00:00:00 2001
From: Jeffrey Wigger <jeffrey.wigger@epfl.ch>
Date: Wed, 9 Mar 2022 17:07:00 +0100
Subject: [PATCH] subsampling fix
---
src/decentralizepy/sharing/FFT.py | 2 +-
src/decentralizepy/sharing/SubSampling.py | 3 +--
2 files changed, 2 insertions(+), 3 deletions(-)
diff --git a/src/decentralizepy/sharing/FFT.py b/src/decentralizepy/sharing/FFT.py
index 1cc8382..1cdf701 100644
--- a/src/decentralizepy/sharing/FFT.py
+++ b/src/decentralizepy/sharing/FFT.py
@@ -4,9 +4,9 @@ import os
from pathlib import Path
from time import time
+import numpy as np
import torch
import torch.fft as fft
-import numpy as np
from decentralizepy.sharing.Sharing import Sharing
diff --git a/src/decentralizepy/sharing/SubSampling.py b/src/decentralizepy/sharing/SubSampling.py
index 1e956cd..5ec0c44 100644
--- a/src/decentralizepy/sharing/SubSampling.py
+++ b/src/decentralizepy/sharing/SubSampling.py
@@ -4,7 +4,6 @@ import os
from pathlib import Path
import torch
-import numpy as np
from decentralizepy.sharing.Sharing import Sharing
@@ -202,7 +201,7 @@ class SubSampling(Sharing):
m["seed"] = seed
m["alpha"] = alpha
- m["params"] = subsample.numpy().astype(np.int32)
+ m["params"] = subsample.numpy()
# logging.info("Converted dictionary to json")
self.total_data += len(self.communication.encrypt(m["params"]))
--
GitLab