search

facebookarchive / fb.resnet.torch

Torch implementation of ResNet from http://arxiv.org/abs/1512.03385 and training scripts
Other
2.29k stars 664 forks source link

I want to retrain a resnext_101_32*4d.Is it possible? #207

Open lyl472324464 opened 5 years ago

lyl472324464 commented 5 years ago

I got this error. dooncloud@dooncloud-System-Product-Name:~/ResNeXt-master$ th main.lua -retrain pretrained/resnext_101_32x4d.t7 -data dooncloud_model/ -resetClassifier true -nClasses 6 Found Environment variable CUDNN_PATH = /usr/local/cuda/lib64/libcudnn.so.5Loading model from file: pretrained/resnext_101_32x4d.t7 => Replacing classifier with 6-way classifier /home/dooncloud/torch/install/bin/luajit: ...ooncloud/torch/install/share/lua/5.1/threads/threads.lua:183: [thread 2 callback] bad argument #2 to '?' (dimension 1 out of range of 0D tensor at /home/dooncloud/torch/pkg/torch/generic/Tensor.c:19)