InvalidArgumentError: Nan in summary histogram for: stn_block_1_in/summaries/histogram_theta

[louisgry]

Hello, have you ever encountered this problem?

tensorflow.python.framework.errors_impl.InvalidArgumentError: Nan in summary histogram for: stn_block_1_in/summaries/histogram_theta
         [[{{node stn_block_1_in/summaries/histogram_theta}}]]

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "main.py", line 82, in <module>
    model_train(PeMS, blocks, args)
    summary, _ = sess.run([merged, train_op], feed_dict={x: x_batch[:, 0:n_his + 1, :, :], keep_prob: 1.0})