search

kuangliu / pytorch-retinanet

RetinaNet in PyTorch
992 stars 250 forks source link

encoder #4

Closed njtuzzy closed 7 years ago

njtuzzy commented 7 years ago

@kuangliu , i am using pytorch 0.12, but i find that in DataEncoder class, the encoder function, "boxes=boxes[max_idx]" can not work, do you mean that this operation can enlarge the size of boxes tensor to anchor size? I can not use this operation due to the different version??

Thanks

kuangliu commented 7 years ago

Update to PyTorch 0.2.0 with advanced indexing support.

njtuzzy commented 7 years ago

Glad to know it, thanks

kuangliu commented 7 years ago

Close this now, feel free to reopen.