Glow: [ImageClassifier] Add option to calculate softmax from the output of the model

Created on 31 Oct 2018  路  2Comments  路  Source: pytorch/glow

Inference models may skip the final softmax op of the network. This means the probabilities printed by the ImageClassifier are incorrect, as it expects softmax to be the final op. We could calculate the softmax in the ImageClassifier itself. This would allow the user to see the probabilities even if the network is missing it. This would likely make sense as a command line option.

good first issue

Most helpful comment

To have the problem description self contained, here is the semantic of softmax:
res_i = exp(input_i) / (exp(input_0) + ... + exp(input_N)).

All 2 comments

For testing purposes, you could use an existing model from Glow model zoo: densenet121

To have the problem description self contained, here is the semantic of softmax:
res_i = exp(input_i) / (exp(input_0) + ... + exp(input_N)).

Was this page helpful?
0 / 5 - 0 ratings

Related issues

tkclimb picture tkclimb  路  4Comments

s-peryt picture s-peryt  路  3Comments

tlepley-cadence picture tlepley-cadence  路  4Comments

opti-mix picture opti-mix  路  4Comments

dati91 picture dati91  路  3Comments