Skip to content

Commit 0f7bf4d

Browse files
committed
Fix TransferLearning.Test null bug.
1 parent a432807 commit 0f7bf4d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/python/subclassing.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -65,7 +65,7 @@ def call(self, x, is_training=False):
6565
x = self.maxpool2(x)
6666
x = self.flatten(x)
6767
x = self.fc1(x)
68-
x = self.dropout(x, training=is_training)
68+
x = self.dropout(x)
6969
x = self.out(x)
7070
if not is_training:
7171
# tf cross entropy expect logits without softmax, so only

0 commit comments

Comments
 (0)