change firstCnn

Author: solawen

firstCnn.py

@@ -46,6 +46,7 @@ def forward(self, x):
 classes = ('plane', 'car', 'bird', 'cat', 'deer',
            'dog', 'frog', 'horse', 'ship', 'truck')
 net = Net()
+net = nn.DataParallel(net)
 net.cuda()
 criterion = nn.CrossEntropyLoss()
 optimizer = optim.SGD(net.parameters(), lr=0.001, momentum=0.9)