search

cl2227619761 / AttFPN-Ovarian-Cancer

MIT License
8 stars 3 forks source link

BUGs repaired for training in GPU #3

Open Chrisa142857 opened 3 years ago

Chrisa142857 commented 3 years ago

ERROR in training with wrong setting of torch.tensor() codes. I added to(device) for each torch.tensor() in FasterRCNN codes.