imagenet-multiGPU.torch icon indicating copy to clipboard operation
imagenet-multiGPU.torch copied to clipboard

Wrong model definition

Open Atcold opened this issue 10 years ago • 2 comments

googlenet.lua does not reflect GoogLeNet architecture (defined in Going Deeper with Convolutions).

In GoogLeNet, inception modules have 2 layers only with 3x3 and 5x5 convolution and 3x3 max-pooling, plus 1x1 dimension matching convolutions.

What is googlenet.lua implementing?

I think there is a little of confusion with the models. From Szegedy we have:

  • GoogLeNet :x:
  • BN-GoogLeNet
  • BN-Inception
  • Inception-v2
    • Vanilla
    • Label Smoothing
    • Factorized 7 × 7
    • BN-auxiliary
  • Inception-v3

and GoogLeNet is not the implemented architecture.

Atcold avatar Feb 29 '16 20:02 Atcold

Here there is a better (more correct) version, but still it is not GoogLeNet.

Atcold avatar Feb 29 '16 21:02 Atcold

Here there is a reliable implementation -> https://github.com/e-lab/imagenet-multiGPU.torch/blob/trueGoogLeNet/models/googlenetFixed_cudnn.lua

Atcold avatar Mar 22 '16 20:03 Atcold