Fixed an error in tensor modification that resulted in backward pass crashing for some networks.

Author: gabikadlecova

nasbench_pytorch/model/model.py

@@ -170,7 +170,7 @@ def forward(self, x):
                 outputs = torch.cat(out_concat, 1)
 
             if self.matrix[0, self.num_vertices-1]:
-                outputs += self.last_inop(tensors[0])
+                outputs = outputs + self.last_inop(tensors[0])
 
             #if self.matrix[0, self.num_vertices-1]:
             #    out_concat.append(self.input_op[self.num_vertices-1](tensors[0]))

setup.py

@@ -2,7 +2,7 @@
 
 setuptools.setup(
     name='nasbench_pytorch',
-    version='1.2.1',
+    version='1.2.2',
     license='Apache License 2.0',
     author='Romulus Hong, Gabriela Suchopárová',
     packages=setuptools.find_packages()