並列にモデルを並べる方法
Q&A
Closed
解決したいこと
pytorchで作ったニューラルネットワークを並列に並べて巨大なニューラルネットワークを作る方法
内容
初心者でよくわかっていなくて申し訳ないのですが、
pytorchにおいて、事前に学習済みの4層の3つのモデルを並列に繋げて、最終層に3→1の層を入れることで、一つの巨大なニューラルネットワークを作り、その上で再学習させたいと考えています。(アンサンブルしたいというようなイメージです。)
しかし、方法がわかりません。どのようなコードを組めば良いのでしょうか。
0 likes