pytorch-yolo-v3 icon indicating copy to clipboard operation
pytorch-yolo-v3 copied to clipboard

use interpolate instead of upsample?

Open richipower opened this issue 6 years ago • 1 comments
trafficstars

Hello i am trying to implement it also in pytorch following your code.. Do you know how can i change the process of the upsample nn.Module created to use interpolate instead of upsample? i would like to get rid of the warning and make a benchmarking speed/performance to see which one is better

thanks!

richipower avatar Feb 21 '19 12:02 richipower

you can custom module ,and place interpolate in forward function.

rongxuanhong avatar Feb 22 '19 04:02 rongxuanhong