我是火炬新手，我已torch7在Ubuntu14.04和cuda8.0上安装cudnn5.1。我尝试过使用mnist和cifar10并且它很好。我使用的代码train-on-digit-classifier和train-on-cifar位于GitHub。
我在使用GPU进行培训之前添加了以下代码。
cutorch.setDevice(1)
model = model:cuda()
criterion = criterion:cuda()
trainData.data = trainData.data:cuda()
trainData.labels = trainData.labels:cuda()
testData.data = testData.data:cuda()
testData.labels = testData.labels:cuda()
它可以运行，但每个时代的正确始终相同。 谁能告诉我哪里出错了？