浏览代码

update

/develop/wb
Ruo-Ping Dong 4 年前
当前提交
4a2512f3
共有 2 个文件被更改,包括 3 次插入3 次删除
  1. 2
      ml-agents/mlagents/trainers/learn.py
  2. 4
      ml-agents/mlagents/trainers/stats.py

2
ml-agents/mlagents/trainers/learn.py


:param run_seed: Random seed used for training.
:param run_options: Command line arguments for training.
"""
#wandb.init(project="mlagent-cloud-profiling", sync_tensorboard=True)
# wandb.init(project="mlagent-cloud-profiling", sync_tensorboard=True)
with hierarchical_timer("run_training.setup"):
checkpoint_settings = options.checkpoint_settings
env_settings = options.env_settings

4
ml-agents/mlagents/trainers/stats.py


self.base_dir: str = base_dir
self._clear_past_data = clear_past_data
with open('wandb_API', 'r') as f:
with open("wandb_API") as f:
os.environ['WANDB_API_KEY'] = api_key
os.environ["WANDB_API_KEY"] = api_key
wandb.init(project="mlagent-cloud-profiling")
wandb.tensorboard.patch(pytorch=True, tensorboardX=True)

正在加载...
取消
保存