我的模型中有多个深度神经网络,并希望它们具有相同的输入大小(网络属于不同的类别)。例如,我的模型是:
class Model:
def __init__(self, cfg: DictConfig):
self.net1 = Net1(**cfg.net1_hparams)
self.net2 = Net2(**cfg.net2_hparams)
这里,Net1 和 Net2 有不同的超参数集,但其中的input_size
参数是 Net1 和 Net2 共享的,必须匹配,即 cfg.net1_hparams.input_size == cfg.net2_hparams.input_size
。
我可以在父级定义 input_size:cfg.input_size
并手动将它们传递给 Net1 和 Net2。但是,我希望每个 Net 的 hparams-configs 都是完整的,以便以后我可以只使用cfg.net1_hparams
.
在 hydra 中是否有实现这一目标的好方法?