【发布时间】:2016-12-30 00:31:49
【问题描述】:
我将构建一个具有多个输出层架构的神经网络。更具体地说,它旨在在一系列卷积层之上构建并行程序。一个分支是计算分类结果(softmax-like);另一种是得到回归结果。但是,我一直在设计模型以及选择损失函数(标准)。
我。我应该使用火炬容器nn.Parallel() 或nn.Concat() 作为conv 层顶部的分支层(nn.Sequential())吗?除了数据格式有什么不同。
二。由于输出数据,分类损失函数和回归损失函数将线性组合。我想知道是否要针对确定的容器选择nn.MultiCriterion() 或nn.ParallelCriterion()。或者我必须自定义一个新的标准类。
三。做过类似工作的任何人都可以告诉我,torch 是否需要额外的定制来实施 backprop 进行训练。我担心火炬容器的数据结构问题。
【问题讨论】:
标签: neural-network deep-learning convolution torch