-
Monica Alfaro authored
* modified FasterRCNN to accept negative samples * remove debug lines * Change torch.zeros_like to torch.zerros * Add unit tests * take the `device` into account Co-authored-by:Francisco Massa <fvsmassa@gmail.com>
e75b4973
* modified FasterRCNN to accept negative samples
* remove debug lines
* Change torch.zeros_like to torch.zerros
* Add unit tests
* take the `device` into account
Co-authored-by: Francisco Massa <fvsmassa@gmail.com>