machine learning - training - Instalando la versión de cunn de GitHub y usando el criterio de multilabelmargin

pytorch mnist tutorial

Question

Necesito usar el criterio de marcado multilabo en torch7 para entrenar una CNN. No es compatible con el cunn actual. He intentado instalar una versión diferente de cunn, como aquí: fuente

git clone https://github.com/clementfarabet/cunn -b master
cd cunn
Luarocks make rocks/cunn-scm-1.rockspec

Sin embargo, cuando lo hago, la antorcha deja de funcionar y aparece el error:

make[1]: *** [CMakeFiles/cunn.dir/all] Error 2
make: *** [all] Error 2

Posteriormente necesité reinstalar torch7 usando ezinstall. ¿Cómo puedo instalar correctamente la versión de cunn requerida, sin afectar la instalación de mi soplete y, por lo tanto, usar el marcador múltiple?