提交 456f3676 authored 作者: Benjamin Scellier's avatar Benjamin Scellier

file theano/compile/monitormode.py

上级 b7550758
from __future__ import absolute_import, print_function, division from __future__ import absolute_import, print_function, division
# Note: this code was initially copied from the 'pyutools' package by its # Note: this code was initially copied from the 'pyutools' package by its
# original author, and re-licensed under Theano's license. # original author, and re-licensed under Theano's license.
import numpy import numpy as np
import theano import theano
from theano.compile.mode import Mode from theano.compile.mode import Mode
...@@ -93,8 +93,8 @@ class MonitorMode(Mode): ...@@ -93,8 +93,8 @@ class MonitorMode(Mode):
def detect_nan(i, node, fn): def detect_nan(i, node, fn):
for output in fn.outputs: for output in fn.outputs:
if (not isinstance(output[0], numpy.random.RandomState) and if (not isinstance(output[0], np.random.RandomState) and
numpy.isnan(output[0]).any()): np.isnan(output[0]).any()):
print('*** NaN detected ***') print('*** NaN detected ***')
theano.printing.debugprint(node) theano.printing.debugprint(node)
print('Inputs : %s' % [input[0] for input in fn.inputs]) print('Inputs : %s' % [input[0] for input in fn.inputs])
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论