Problem about prediction score when testing an image
Abierto
- Lenguaje dominante
- Lua
- Estrellas
- 872
- Forks
- 235
- Métricas de merge de PR
- Sin PR fusionados en 30 d
Descripción
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??
Guía de contribución
No hay ninguna guía de contribución indexada para este repositorio
Evaluación
Este issue todavía no se ha evaluado.