在Detectron2中使用DefaultTrainer训练时如何更改模型参数保存位置

Question

我的代码如下：


cfg = get_cfg()

...
trainer = DefaultTrainer(cfg)
trainer.resume_or_load(resume=False)
trainer.train()

训练结束后，模型自动保存在

./output/model_final.pth

中。我找到了保存模型的文件夹，但是找不到更改模型文件名的接口

如何更改保存的文件名？如果您能帮助我，我将非常感激

Answer 1

我花了很长时间试图解决这个问题，但仍然没有实现。但我使用的解决方法是仅更改路径本身而不是文件名。

cfg.OUTPUT_DIR =“./path/to/”

所以我得到了output/hands/model_final.pth 和output/feet/model_final.pth 而不是不同的文件名。

Answer 2

以防其他人可能发现这个问题。

添加

_C.OUTPUT_DIR

将更改写入模型文件的目录。