removing not needed to list; set_num_threads to 2;

src/decentralizepy/node/Node.py

@@ -3,6 +3,7 @@ import json
 import logging
 import os
 
+import torch
 from matplotlib import pyplot as plt
 
 from decentralizepy import utils
@@ -420,6 +421,8 @@ class Node:
             Other arguments
 
         """
+        torch.set_num_threads(2)
+        torch.set_num_interop_threads(1)
         self.instantiate(
             rank,
             machine_id,

src/decentralizepy/sharing/PartialModel.py

@@ -155,9 +155,9 @@ class PartialModel(Sharing):
             if not self.dict_ordered:
                 raise NotImplementedError
 
-            m["indices"] = G_topk.numpy().tolist()
+            m["indices"] = G_topk.numpy()
 
-            m["params"] = T_topk.numpy().tolist()
+            m["params"] = T_topk.numpy()
 
             assert len(m["indices"]) == len(m["params"])
             logging.info("Elements sending: {}".format(len(m["indices"])))