提交 e6dc5e8d authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Allow floatX=float16

上级 bce0954f
...@@ -18,13 +18,16 @@ def floatX_convert(s): ...@@ -18,13 +18,16 @@ def floatX_convert(s):
return "float32" return "float32"
elif s == "64": elif s == "64":
return "float64" return "float64"
elif s == "16":
return "float16"
else: else:
return s return s
AddConfigVar('floatX', AddConfigVar('floatX',
"Default floating-point precision for python casts", "Default floating-point precision for python casts",
EnumStr('float64', 'float32', convert=floatX_convert,), EnumStr('float64', 'float32', 'float16',
) convert=floatX_convert,),
)
AddConfigVar('warn_float64', AddConfigVar('warn_float64',
"Do an action when a tensor variable with float64 dtype is" "Do an action when a tensor variable with float64 dtype is"
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论