From f80fed3caa8e6136887ebba613111e6e516d223e Mon Sep 17 00:00:00 2001
From: Milos Vujasinovic <milos.vujasinovic@epfl.ch>
Date: Mon, 24 Oct 2022 13:51:02 +0200
Subject: [PATCH] Choco compression fix
---
src/decentralizepy/sharing/Choco.py | 6 +++---
1 file changed, 3 insertions(+), 3 deletions(-)
diff --git a/src/decentralizepy/sharing/Choco.py b/src/decentralizepy/sharing/Choco.py
index 05de209..bf1d77f 100644
--- a/src/decentralizepy/sharing/Choco.py
+++ b/src/decentralizepy/sharing/Choco.py
@@ -310,9 +310,9 @@ class Choco(Sharing):
model,
dataset,
log_dir,
- compress=False,
- compression_package=None,
- compression_class=None
+ compress,
+ compression_package,
+ compression_class
)
self.step_size = step_size
self.alpha = alpha
--
GitLab