Commit eea85d10 authored by suily's avatar suily
Browse files

添加推理结果

parent 67a1d360
...@@ -154,6 +154,7 @@ python test.py ...@@ -154,6 +154,7 @@ python test.py
<div align=center> <div align=center>
<img src="./doc/picsum.jpg"/> <img src="./doc/picsum.jpg"/>
</div> </div>
``` ```
dcu推理结果: dcu推理结果:
0.73861 : alp 0.73861 : alp
......
...@@ -19,7 +19,7 @@ import tensorflow_datasets as tfds ...@@ -19,7 +19,7 @@ import tensorflow_datasets as tfds
import time import time
# import tensorflow as tf # import tensorflow as tf
'''显示可用设备gpu的数量。''' '''测试dcu/gpu'''
from jax.lib import xla_bridge from jax.lib import xla_bridge
jax_test=xla_bridge.get_backend().platform jax_test=xla_bridge.get_backend().platform
if not (jax_test=='gpu'): if not (jax_test=='gpu'):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment