search

bingykang / Fewshot_Detection

Few-shot Object Detection via Feature Reweighting
https://arxiv.org/abs/1812.01866
526 stars 111 forks source link

Base Training error #31

Open Miracle-hpf opened 4 years ago

Miracle-hpf commented 4 years ago

Sorry for troubling you.I follow your instruction and when I run train_meta.py, a runtimeerror occured as follow: 1

Could you tell me how to solve it?

eghouti commented 4 years ago

you can solve it as follows tmp = buf[start:start+num_w] tmp = tmp.view(conv_model.weight.shape) conv_model.weight.data.copy(tmp)

Miracle-hpf commented 4 years ago

you can solve it as follows tmp = buf[start:start+num_w] tmp = tmp.view(conv_model.weight.shape) conv_model.weight.data.copy(tmp)

Thanks for your reply!I reduce the torch version to 0.3.1 and now can run.

zhongdajian commented 4 years ago

you can solve it as follows tmp = buf[start:start+num_w] tmp = tmp.view(conv_model.weight.shape) conv_model.weight.data.copy(tmp)

Hello, i have some questions about the training samplse, can you help me? 'train = ./voc_train.txt' in the 'cfg/metayolo.data', it contains 20 categories, so it contains the novel categories. Should we use 15 base categories instead?