未验证 提交 63f1e6bd 编写于 作者: J Jiabin Yang 提交者: GitHub

Remove additional warnning in layer.to (#36700)

* remove additional warnning in layer.to

* remove additional warnning in layer.to

* remove additional warnning in layer.to

* remove additional warnning in layer.to

* remove additional warnning in layer.to
上级 5119428e
......@@ -1590,7 +1590,10 @@ class Layer(core.Layer):
return new_t
self._apply(transform, device, dtype, blocking)
with warnings.catch_warnings():
warnings.filterwarnings("ignore", category=UserWarning)
self._apply(transform, device, dtype, blocking)
self._dtype = dtype
# [aliases] Compatible with old method names
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册