pytorch-yolo-v3
pytorch-yolo-v3 copied to clipboard
use interpolate instead of upsample?
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!
you can custom module ,and place interpolate in forward function.