モデルに複数のディープ ニューラル ネットワークがあり、それらの入力サイズを同じにしたい (ネットワークのクラスが異なる)。たとえば、私のモデルは次のとおりです。
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 を定義し、Net1cfg.input_size
と Net2 の両方に手動で渡すことができます。しかし、各ネットの hparams-configs を完成させて、後でcfg.net1_hparams
.
ヒドラでこれを達成する良い方法はありますか?