Problem about prediction score when testing an image
Offen
- Vorherrschende Sprache
- Lua
- Sterne
- 872
- Forks
- 235
- PR-Merge-Kennzahlen
- Keine gemergten PRs in 30 T.
Beschreibung
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??
Beitragsleitfaden
Für dieses Repository ist kein Beitragsleitfaden indexiert
Bewertung
Dieses Issue wurde noch nicht bewertet.