Problem with training Tiny YOLOv2 on VOC

[jefersonf]

Hi @andy-yun

I'm trying to train the tiny yolov2 on VOC, but I got the following error:

...
File "/home/jeferson/.local/lib/python3.5/site-packages/torch/nn/modules/module.py", line 518, in __getattr__
    type(self).__name__, name))
AttributeError: 'Darknet' object has no attribute 'module

When I tried to run this.

python3 train.py -d cfg/voc.data -c cfg/tiny-yolo-voc.cfg -w yolov2-tiny-voc.weights 

Could you guess what the problem is? I'm using pytorch 0.4.1

[andy-yun]

Your log shows that ngpus>1. Please check the voc.data with gpus=

_{Sent with GitHawk}