iicd/xautodl
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix typos in GDAS reduction cell

Browse Source
This commit is contained in:
D-X-Y 2020-02-24 15:26:43 +11:00
parent 4e322bb9c7
commit e0f131caac
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
lib/models/cell_operations.py
View File

@ -284,7 +284,8 @@ class GDAS_Reduction_Cell(nn.Module):
if self.training and drop_prob > 0.: if self.training and drop_prob > 0.:
X0, X1 = drop_path(X0, drop_prob), drop_path(X1, drop_prob) 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) X3 = self.ops2[1] (s1)
if self.training and drop_prob > 0.: if self.training and drop_prob > 0.:
X2, X3 = drop_path(X2, drop_prob), drop_path(X3, drop_prob) X2, X3 = drop_path(X2, drop_prob), drop_path(X3, drop_prob)