Fix typos in GDAS reduction cell
This commit is contained in:
parent
4e322bb9c7
commit
e0f131caac
@ -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)
|
||||
|
Loading…
Reference in New Issue
Block a user