Open 9thDimension opened 7 years ago
I meet the same situation, are you solve it ?~
Although I rewrite the VOCDataset(but the main code is the same as longcw write). Maybe, the problem you meet is the same as me.
The problem is mainly by the cfg.train_batch_size
. When I set it equal to 1, the problem is appeared.
Using the default cfg.train_batch_size (which is 16), the problem is gone. ~ (I don't check the reason behind it, but when i see the self.bbox_loss become very large after several iteration. --- you can debug by yourself).
I've been interested in using YOLO architecture for an object detection task. As a first step I decided to clone this repo and run the examples. Upon running the
train.py
I get these results...Note that I had to compile the 'roi_pooling' and 'reorg' modules (not 100% sure what these are for) with slightly different flags (
-arch=sm_30
) to match my laptop's GPU. In the future I intend to use AWS for full-scale training. Also I'm a new to the PyTorch framework.