We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 01f0f08 commit 6a85dc6Copy full SHA for 6a85dc6
neural_structured_learning/research/gam/models/cnn.py
@@ -320,7 +320,7 @@ def get_loss(self,
320
loss = tf.losses.softmax_cross_entropy(targets, predictions)
321
322
# Weight decay loss.
323
- if weight_decay:
+ if weight_decay is not None:
324
for var in reg_params.values():
325
loss += weight_decay * tf.nn.l2_loss(var)
326
return loss
0 commit comments