将if __name__ == '__main__': tf.app.run()改为tensorflow2.0版本的API
时间: 2023-07-21 16:46:05 浏览: 175
Tensorflow中的降维函数tf.reduce_*使用总结
在 TensorFlow 2.0 中,可以使用以下代码替换原来的代码:
```python
if __name__ == '__main__':
import tensorflow.compat.v1 as tf
tf.disable_v2_behavior()
tf.app.run()
```
这里我们先将 TensorFlow 的默认版本设置为 1.x(也就是 `tf.compat.v1`),然后再运行 `tf.app.run()`。这样就可以在 TensorFlow 2.0 中使用原来的代码了。
阅读全文