2

モデルに複数のディープ ニューラル ネットワークがあり、それらの入力サイズを同じにしたい (ネットワークのクラスが異なる)。たとえば、私のモデルは次のとおりです。

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.

ヒドラでこれを達成する良い方法はありますか?

4

1 に答える 1