Browse Source

Fix TransferLearning.Test null bug.

tags/v0.100.4-load-saved-model
Haiping Chen 2 years ago
parent
commit
0f7bf4d6a6
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      src/python/subclassing.py

+ 1
- 1
src/python/subclassing.py View File

@@ -65,7 +65,7 @@ class ConvNet(Model):
x = self.maxpool2(x)
x = self.flatten(x)
x = self.fc1(x)
x = self.dropout(x, training=is_training)
x = self.dropout(x)
x = self.out(x)
if not is_training:
# tf cross entropy expect logits without softmax, so only


Loading…
Cancel
Save