tensorflow结果可重复
文章目录
最近在学吴恩达的深度学习课程,在参考别人的代码的时候,发现在CPU多线程情况下和GPU运行的情况下,同样的代码、数据,每次训练出来的结果都不一样,这还怎么调参,于是stackoverflow走起。
开发环境
- Windows10 64位
- python 3.6
- tensorflow 1.14
- keras 2.2.5
- CUDA 10.0
- cudnn for CUDA 10.0
解决过程
用CPU跑
# Seed value
# Apparently you may use different seed values at each stage
seed_value= 0
# 1. Set the `PYTHONHASHSEED` environment variable at a fixed value
import os
os.environ[