浏览代码

Added growth parameter to stop failing with allocation under windows for #277 (#278)

/tag-0.2.1c
Arthur Juliani 7 年前
当前提交
ce2ce437
共有 1 个文件被更改,包括 4 次插入2 次删除
  1. 6
      python/ppo.py

6
python/ppo.py


return None
else:
return None
with tf.Session() as sess:
config = tf.ConfigProto()
config.gpu_options.allow_growth = True
with tf.Session(config=config) as sess:
# Instantiate model parameters
if load_model:
print('Loading Model...')

正在加载...
取消
保存