我点击运行wandb_denoisekt_train.py文件显示 NoneType,
我在init_model.py上加上了 elif model_name == "denoisekt":
model = DenoiseKT(data_config["num_c"],data_config["num_q"], **model_config, emb_type=emb_type, emb_path=data_config["emb_path"]).to(device)
加了这两行代码后能训练模型了,请问这行代码正确吗?