hyl-g comments

Repositories
Issues
Comments

Results 2 comments of


                                            hyl-g

Sudden jump in loss from Epoch 0 to Epoch 1 when training OpenImages

Will do.

Retraining RuntimeError: expect torch.LongTensor but found torch.IntTensor

The problem was solved by casting labels in functions train() and test() to torch.LongTensor. See the diff below: [diff_train_ssd.txt](https://github.com/qfgaohao/pytorch-ssd/files/3606244/diff_train_ssd.txt)