浏览代码

[skip ci] minor formatting change

/distributed-training
Anupam Bhatnagar 5 年前
当前提交
b3c2d431
共有 1 个文件被更改,包括 2 次插入1 次删除
  1. 3
      ml-agents/mlagents/trainers/optimizer/tf_optimizer.py

3
ml-agents/mlagents/trainers/optimizer/tf_optimizer.py


if hvd is not None:
adam_optimizer = tf.train.AdamOptimizer(
learning_rate=learning_rate, name=name)
learning_rate=learning_rate, name=name
)
horovod_optimizer = hvd.DistributedOptimizer(adam_optimizer)
else:
adam_optimizer = tf.train.AdamOptimizer(

正在加载...
取消
保存