budaLi

pytorch to caffe

4

Platform (like ubuntu 16.04/win10): ubuntu 18.04 Python version: 3.6 Source framework with version (like Tensorflow 1.4.1 with GPU): pytorch 0.4.0 torchvision 0.2.1 when I trans pth to caffe magic_number =...

feature extract error

5


模型输入为 30个数字，经过多层全连接，每层全连接加了relu，输出的结果不对 debug看好像不会对torch的这种网络做激活层的输出，有什么好的修改办法吗。 demo网络如下。 class SmallNet(nn.Module): def __init__(self): super(SmallNet, self).__init__() self.fc1 = nn.Linear(30, 128) self.fc2 = nn.Linear(128, 64) self.fc3 = nn.Linear(64, 2) def forward(self, x): x = torch.relu(self.fc1(x)) x =...