您最多选择25个主题
主题必须以中文或者字母或数字开头，可以包含连字符 (-)，并且长度不得超过35个字符
34 行
933 B
34 行
933 B
import logging
|
|
|
|
|
|
def warnings_as_errors(log_record):
|
|
# Raise deprecated warnings as exceptions.
|
|
if log_record.levelno == logging.WARNING and "deprecated" in log_record.msg:
|
|
merged = log_record.getMessage()
|
|
raise RuntimeError(merged)
|
|
return True
|
|
|
|
|
|
# TODO only enable this with a environment variable
|
|
if False:
|
|
logging.getLogger("tensorflow").addFilter(warnings_as_errors)
|
|
|
|
try:
|
|
import tensorflow.compat.v1 as tf
|
|
except ImportError:
|
|
import tensorflow as tf
|
|
|
|
# TODO better version check, this will do for now though
|
|
is_tf2 = tf.__version__ == "2.0.0"
|
|
if is_tf2:
|
|
tf_variance_scaling = tf.initializers.variance_scaling
|
|
tf_flatten = tf.layers.flatten
|
|
tf_rnn = tf.nn.rnn_cell
|
|
|
|
tf.disable_v2_behavior()
|
|
else:
|
|
import tensorflow.contrib.layers as c_layers
|
|
|
|
tf_variance_scaling = c_layers.variance_scaling_initializer
|
|
tf_flatten = c_layers.flatten
|
|
tf_rnn = tf.contrib.rnn
|