oyxhust / ssd-text_detection

A modified SSD model for text detection
91 stars 41 forks source link

RuntimeError: simple_bind error. Arguments #6

Closed Issuenate closed 4 years ago

Issuenate commented 6 years ago

Hi oyxhust, During training the program, i ran into this problem, do you have any idea about how to deal with it? Thanks for your time!

issue@thisiscaffe:~/Desktop/text-detection$ python train_synthtext.py loading data... finish loading... INFO:root:Start training with (gpu(0)) from pretrained model /home/issue/Desktop/text-dete ction/model/vgg16_reduced [15:41:59] src/nnvm/legacy_json_util.cc:190: Loading symbol saved by previous version v0.8 .0. Attempting to upgrade... [15:41:59] src/nnvm/legacy_json_util.cc:198: Symbol successfully upgraded! INFO:root:Freezed parameters: [conv1_1_weight,conv1_1_bias,conv1_2_weight,conv1_2_bias,con v2_1_weight,conv2_1_bias,conv2_2_weight,conv2_2_bias] Traceback (most recent call last): File "train_synthtext.py", line 76, in args.lr_refactor_ratio, args.monitor, args.log_file) File "/home/issue/Desktop/text-detection/train/train_net.py", line 274, in train_net monitor=monitor) File "/usr/local/lib/python2.7/dist-packages/mxnet/module/base_module.py", line 460, in fit for_training=True, force_rebind=force_rebind) File "/usr/local/lib/python2.7/dist-packages/mxnet/module/module.py", line 429, in bind state_names=self._state_names) File "/usr/local/lib/python2.7/dist-packages/mxnet/module/executor_group.py", line 264, in init self.bind_exec(data_shapes, label_shapes, shared_group) File "/usr/local/lib/python2.7/dist-packages/mxnet/module/executor_group.py", line 360, in bind_exec shared_group)) File "/usr/local/lib/python2.7/dist-packages/mxnet/module/executor_group.py", line 638, in _bind_ith_exec shared_buffer=shared_data_arrays, **input_shapes) File "/usr/local/lib/python2.7/dist-packages/mxnet/symbol/symbol.py", line 1518, in simp le_bind raise RuntimeError(error_msg)

RuntimeError: simple_bind error. Arguments: data: (32, 3L, 300L, 300L) label: (32, 63L, 5L) [15:42:01] src/storage/storage.cc:118: Compile with USE_CUDA=1 to enable GPU usage