Problem about prediction score when testing an image
Open
- Dominant language
- Lua
- Stars
- 872
- Forks
- 235
- PR merge metrics
- No merged PRs in 30d
Description
When I used the pre-trained module to test an image, I wrote codes as below
```
predictions = model:forward(img:cuda())
print(predictions:exp())
```
and then I got
`[torch.CudaTensor of size 4x1000]`
,which means the vector about the final prediction score is 4*1000.
However we know the last layer is
`nn.View(1000)`
My test result have got three extra dimension, WHY??
Contributor guide
No contributing guide indexed for this repository
Assessment
This issue has not been assessed yet.