omegaconf.errors.ConfigAttributeError: Missing key training_model

[XinyingZheng]

error info as follows:
omegaconf.errors.ConfigAttributeError: Missing key training_model
full_key: training_model
object_type=dict

run command:
python bin/predict.py +config=/code/FishDreamer/configs/training/FishDreamer.yaml model.path="/code/FishDreamer/fishdreamer.ckpt" indir=/data/new_data/fisheye_imgs/ outdir=./
你好，想直接跑推理看效果，fishdreamer.ckpt这个预训练模型对应的配置config文件是哪个

[Tiankong-1]

After two days of testing, FishDreamer's training code and inference code cannot be implemented, or it may be due to incorrect environment configuration. If what I said is not accurate, please contact me. At the same time, it is also hoped that the author can provide a more detailed environmental configuration plan to achieve the effectiveness of the paper.

[bluemoonwencong]

+1