forresti / SqueezeNet

SqueezeNet: AlexNet-level accuracy with 50x fewer parameters
BSD 2-Clause "Simplified" License
2.17k stars 723 forks source link

The SqueezeNet deploy.caffemodel files have all 0.0 weight and bias data #56

Closed jnorwood closed 6 years ago

jnorwood commented 6 years ago

The comments on the SqueezeNet models indicate that the .caffemodel files include the training data. The deploy.caffemodel files' weights and biases are 0.0. Is this intentional? It wasn't evident from the comments. I see the squeezenet_v1.1.caffemodel does have non-zero weights and biases.

Look at any of the parameter data

import caffe
import numpy
nnet = caffe.Net("deploy.prototxt",1, weights= "deploy.caffemodel")
net.params['conv1'][0].data[...] # weights
net.params['conv1'][1].data[...] # biases 
jnorwood commented 6 years ago

The train_val.prototxt does contain appropriate weights and biases, which solves my problem. I'll close this.