Update models.py

sileod · Mar 19, 2023 · feca18c · feca18c
1 parent ba37176
commit feca18c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/tasknet/models.py b/src/tasknet/models.py
@@ -219,7 +219,7 @@ def forward(self, task, **kwargs):
         return y
 
     def factorize(self, base_index=0, tasks=[],labels=[]):
-        m_i = self.task_models_list[base_index]
+        m_i = copy.deepcopy(self.task_models_list[base_index])
         classifiers = torch.nn.ModuleList([a.classifier for a in self.task_models_list])
         if hasattr(m_i,'auto'):
             del m_i.auto