I trained pixor with a batch-size 1 and run for around35 epoch. Loss was exponentially decreasing.
With that trained model when i started evaluated using python main.py --mode=test --name=default or python main.py --mode=val --name=default, iam getting no prediction boxes.
I trained with batchsize1 using 1080Ti since for larger batch_size>=4, the code was crashing.
Kindly help me/suggest me in getting correct prediction boxes. Pls help
ghost
commented
5 years ago
Hi,
I trained pixor with a batch-size 1 and run for around35 epoch. Loss was exponentially decreasing.
With that trained model when i started evaluated using python main.py --mode=test --name=default or python main.py --mode=val --name=default, iam getting no prediction boxes. I trained with batchsize1 using 1080Ti since for larger batch_size>=4, the code was crashing.
Kindly help me/suggest me in getting correct prediction boxes. Pls help