Error trying to run custom caffenet with new data
问题 I just tried to train the provided Caffenet network with my own lmdb file. I changed the fully connected to convolutional layer of depth 4096 and custom frame_size. Here is the code: weight_param = dict(lr_mult=1, decay_mult=1) bias_param = dict(lr_mult=2, decay_mult=0) learned_param = [weight_param, bias_param] batch_size = 256 # 0 means non updating parameters frozen_param = [dict(lr_mult=0)] * 2 def conv_relu(bottom, ks, nout, stride=1, pad=0, group=1, param=learned_param, weight_filler