我目前正在为我的深度学习项目试验 hydra-conf。对于培训,我遵循文档中显示的方式:
- 配置文件的文件夹,一个主要的config.yml文件,在额外的文件夹中有额外的配置组
- 我的主要方法用 @hydra.main(...) 装饰来初始化/加载配置
当我的模型被训练时,我保存了相应的状态字典。
现在的问题是,使用 hydra 进行模型推理的预期方法是什么?我的想法是在加载状态字典之前,使用模型检查点以及 .hydra 文件夹中特定于运行的config.yml文件来获取实例化模型所需的构造函数参数。
有没有更好的方法来实现我的目标?
编辑 1: 当尝试使用 @hydra.main() 方法时,我不知道如何将动态配置文件路径传递给装饰器,因为这将由用户提供(场景:我想对通过在 hydra 创建的 training-run 文件夹中提供配置文件的路径来创建特定模型)