fix get_para_vector function

conflictfree/utils.py

@@ -71,8 +71,8 @@ def apply_para_vector(network:torch.nn.Module,para_vec:torch.Tensor)->None:
     with torch.no_grad():
         start=0
         for par in network.parameters():
-            end=start+par.grad.data.view(-1).shape[0]
-            par.data=para_vec[start:end].view(par.grad.data.shape)
+            end=start+par.data.view(-1).shape[0]
+            par.data=para_vec[start:end].view(par.data.shape)
             start=end
 
 def get_cos_similarity(vector1:torch.Tensor,vector2:torch.Tensor)->torch.Tensor:

setup.py

@@ -15,7 +15,7 @@ def get_install_requires():
 
 setuptools.setup(
     name="conflictfree",
-    version="0.1.2",
+    version="0.1.3",
     author="Qiang Liu, Mengyu Chu, Nils Thuerey",
     author_email="[email protected]",
     description="Official implementation of Conflict-free Inverse Gradients method",