我打算建立一個具有多個輸出層結構的神經網絡。更具體地說,它被設計用來在一系列卷積層之上構建並行程序。一個分支是計算分類結果(softmax-like);另一個是獲得迴歸結果。然而,我堅持設計模型以及選擇損失函數(標準)。火炬實現多輸出層神經網絡
I.我應該使用火炬容器nn.Parallel()
或nn.Concat()
作爲位於頂層的分支圖層(nn.Sequential()
)嗎?除數據格式外,什麼是差異。
二,由於輸出數據,分類損失函數和迴歸損失函數應線性組合。我想知道是否nn.MultiCriterion()
或nn.ParallelCriterion()
被確定的容器。或者我必須自定義一個新的標準類。
三,任何做過類似工作的人都可以告訴我,火炬是否需要額外的定製來實施反向訓練。我擔心火炬容器的數據結構問題。