Some modifications for backward propagation

buttercutter · May 29, 2022 · 94bffd1 · 94bffd1
1 parent 6d86824
commit 94bffd1
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/gdas.py b/gdas.py
@@ -255,7 +255,8 @@ def forward(self, x, types=None):
             edges_results = edges_results.cuda()
 
         for e in range(NUM_OF_MIXED_OPS):
-            edges_results = edges_results + self.edges[e].forward(x, types)
+	    with torch.no_grad():
+            	edges_results = edges_results + self.edges[e].forward(x, types)
 
         return edges_results * DECAY_FACTOR