Young-Sun (Andy) Yun

@zhangguotai I updated the code dataset.py and train.py. Try them. Refer to https://discuss.pytorch.org/t/runtimeerror-invalid-argument-0-sizes-of-tensors-must-match-except-in-dimension-0-got-3-and-2-in-dimension-1/23890/15

Maybe.. But you should apply the excessive data augmentation for single image.

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

python detect.py cfg/yolo_v3.cfg yolov3.weights data/dog.jpg data/coco.names see README.md in the root.

@sdustdk1427 same error to #55 I updated dataset.py and train.py. try the code. Refer to https://discuss.pytorch.org/t/runtimeerror-invalid-argument-0-sizes-of-tensors-must-match-except-in-dimension-0-got-3-and-2-in-dimension-1/23890/15

@sdustdk1427 In that case, you can check the information as follows: in dataset.py, you expand `data = torch.stack([item[0] for item in batch],0)` ``` try: data = torch.stack([item[0] for item in...

@sdustdk1427 If you comment out "def custom_collate", then default collate_fn is used. Then this phenomenon is exactly same to the first condition (without collate_fn). custom_collate function is used for checking...

You can add max_epochs in [net] session.

You try the following in config file [net] max_epochs=10000

cfg/yolo_v3.cfg or similar config file.