search

soumith / imagenet-multiGPU.torch

an imagenet example in torch.
BSD 2-Clause "Simplified" License
402 stars 158 forks source link

Vggbn #66

Closed sinaghs closed 8 years ago

sinaghs commented 8 years ago

Hi, Thanks for the great script. There is a problem in vggbn model, in the classifier part classifier:add(nn.Linear(4096, 1000)) shouldn't be replaced with: classifier:add(nn.Linear(4096, nrclasses))