search

luuuyi / RefineDet.PyTorch

A higher performance PyTorch implementation of Single-Shot Refinement Neural Network for Object Detection
MIT License
235 stars 64 forks source link

sorry,something wrong i don't know how to solve it #20

Closed rw1995 closed 5 years ago

rw1995 commented 5 years ago

图片 =-=please what happen,thx to you

rw1995 commented 5 years ago

i only follow your suggestion put the weight into weights and do the sh

luuuyi commented 5 years ago

@rw1995 I can not get any more infomation with this pic, but I wonder know whether you have multi-gpu in your machine.

rw1995 commented 5 years ago

i only have one GPU，i don't know why=-= @luuuyi when it train on voc2007, itis ok, but in my dataset ,it don't work

luuuyi commented 5 years ago

@rw1995 when you just have one gpu, you should set gpu_num equal to 1, i noticed that your error info mentions nn.DataParallel, you do not need to use it when you just have one gpu.