Tongcheng / DN_CaffeScript

128 stars 58 forks source link

Train with other dataset #14

Open VincentGu11 opened 6 years ago

VincentGu11 commented 6 years ago

Hi Tongcheng,

Thank you for your implementation of Densenet Caffe and I can already train Cifar 10 dataset. However, when it comes to my own dataset, it will be stuck for a long time at here:

I0111 00:30:41.302551 8065 layer_factory.hpp:77] Creating layer Data1 I0111 00:30:41.337936 8065 db_lmdb.cpp:35] Opened lmdb /home/gaia/Dev/caffe/examples/Apollo/lmdb_test/apollo_general18_train_lmdb I0111 00:30:41.356104 8065 net.cpp:86] Creating Layer Data1 I0111 00:30:41.356127 8065 net.cpp:382] Data1 -> Data1 I0111 00:30:41.356150 8065 net.cpp:382] Data1 -> Data2 I0111 00:30:41.357967 8065 data_layer.cpp:48] output data size: 1,3,512,512

After a long time (about 1hour), the process will be killed without any error information. My input size of images is 512*512. I use "train_test_BCBN_C10plusNoBias.prototxt" for my network architecture and I only modified the input and the "num_output" in InnerProduct1 layer. Do you think it is too large? Or I have made some other mistakes? Can you give me some advice? Thanks a lot!