allenai / allenai/XNOR-Net

Problem about prediction score when testing an image

Abierto
#30 1 comentario 0 reacciones 0 asignados Ver en GitHub
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.

Recibe los nuevos issues en tu correo

Un resumen breve de issues de GitHub para principiantes.