Prechádzať zdrojové kódy

Merge pull request #36 from argszero/patch-1

Using a `tf.Tensor` as a Python `bool` is not allowed
Sergio Guadarrama 9 rokov pred
rodič
commit
30004531ab
1 zmenil súbory, kde vykonal 1 pridanie a 1 odobranie
  1. 1 1
      inception/inception/inception_train.py

+ 1 - 1
inception/inception/inception_train.py

@@ -266,7 +266,7 @@ def train(dataset):
 
     # Add histograms for gradients.
     for grad, var in grads:
-      if grad:
+      if grad is not None:
         summaries.append(
             tf.histogram_summary(var.op.name + '/gradients', grad))