New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

1. decouple boxes, labels; 2: cast lables to 64 bits #74

Merged

qfgaohao merged 1 commit into qfgaohao:master from hyl-g:fix_open_images

Sep 14, 2019

hyl-g commented Sep 13, 2019

As the title said, there are two changes:

Decouple boxes and labels in _getitem() to prevent corruption of dataset.
In _read_data(), cast labels to int16 because cross_entropy() expects torch.LongTensor.


          1. decouple boxes, labels; 2: cast lables to 64 bits

4d54172

qfgaohao merged commit 7174f33 into qfgaohao:master

Owner

qfgaohao commented Sep 14, 2019

@hyl-g Thanks a lot!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet