Fix typos in GDAS reduction cell

2020-02-24 15:26:43 +11:00
parent 4e322bb9c7
commit e0f131caac
1 changed files with 2 additions and 1 deletions
--- a/lib/models/cell_operations.py
+++ b/lib/models/cell_operations.py
@@ -284,7 +284,8 @@ class GDAS_Reduction_Cell(nn.Module):
    if self.training and drop_prob > 0.:
      X0, X1 = drop_path(X0, drop_prob), drop_path(X1, drop_prob)

-    X2 = self.ops2[0] (X0+X1)
+    #X2 = self.ops2[0] (X0+X1)
+    X2 = self.ops2[0] (s0)
    X3 = self.ops2[1] (s1)
    if self.training and drop_prob > 0.:
      X2, X3 = drop_path(X2, drop_prob), drop_path(X3, drop_prob)