pytorch 如何为hydra生成的日志文件设置目录,而不是当前的工作目录?

bkhjykvo  于 12个月前  发布在  其他
关注(0)|答案(1)|浏览(203)

我使用的是Hydra 1.2版。我将Hydra的默认工作目录更改为另一个目录。现在我想将Hydra生成的日志文件保存到其他目录中。我如何才能做到这一点?
目前,日志文件保存在当前工作目录,我找不到一种方法来定义一个目录,我可以保存日志文件。

uidvcgyl

uidvcgyl1#

你可以通过在自定义配置文件中覆盖它来配置Hydra运行目录。更多信息请参见这里的文档:https://hydra.cc/docs/configure_hydra/workdir/
下面是一个简化的例子,说明我是如何做到的,其中name是我计划在配置的其他地方使用的东西。

name: benchmark
device: cuda:1
#------------------------- Hydra Config ------------------------#
override hydra/hydra_logging: colorlog
override hydra/job_logging: colorlog
hydra:
  job:
    name: ${name}
  run:
    dir: outputs/${hydra.job.name}/${now:%Y-%m-%d_%H-%M-%S}
  sweep:
    dir: ${hydra.job.name}
    subdir: ${hydra.job.num}

字符串
你可以通过使用HydraConfig导入来测试它是否工作,如下所示:

from hydra.core.hydra_config import HydraConfig

@hydra.main()
def my_app(cfg: DictConfig) -> None:
    print(HydraConfig.get().runtime.output_dir)

相关问题