search

yokattame / SpindleNet

115 stars 36 forks source link

training different parts with one gpu #13

Closed RuaYahya closed 6 years ago

RuaYahya commented 6 years ago

Dear all, I have trained the base model successfully but when I try to train the train_head using train_head.sh I got an error, see the attached photo. I know that the problem with this line GLOG_log_dir=${log} ${CAFFE_DIR}/build/tools/caffe train --solver=${solver} --gpu=0\ --weights=${pretrained_model} I have one gpu (GeForce GTX 1070) installed. I play around change it but unfortunately ,no result.Any advice will be highly appreciated.

sasa

RuaYahya commented 6 years ago

please any advice will be highly appreciated. Thanks

RuaYahya commented 6 years ago

@kotabe-sh @xuyunlu1030 @Smellly Can I know the hardware specifications thats make this code run for you.

Thanks in advance.

RuaYahya commented 6 years ago

@yokattame, please help. I know you may be very busy, but I spend few days trying to solve this error without result. I have train the base_train successfully but when I try the head, it does not work.

Your collaboration is highly appreciated.

RuaYahya commented 6 years ago

Works now. Thanks