From e0f131caac9193de448d6d0af486323049331257 Mon Sep 17 00:00:00 2001 From: D-X-Y <280835372@qq.com> Date: Mon, 24 Feb 2020 15:26:43 +1100 Subject: [PATCH] Fix typos in GDAS reduction cell --- lib/models/cell_operations.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/lib/models/cell_operations.py b/lib/models/cell_operations.py index f9e6a20..ce66794 100644 --- 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)