search

ry / tensorflow-resnet

ResNet model in TensorFlow
MIT License
1.66k stars 625 forks source link

Modification to return logit in inference_small() #7

Open takerum opened 8 years ago

takerum commented 8 years ago

I found the missing line for returning logit in inference_small(). I added the line at the end of inference_small().

JulesGM commented 8 years ago

This should be accepted. I also just stumbled into this, and came to the same conclusion:

inference_small doesn't return anything, yet in the cifar example, it's (non-existant) value is assigned to a variable. I'm pretty sure it's just that "return" is missing on the last line of inference_small, before inference_small_config(x, c).