fidler-lab / polyrnn-pp-pytorch

PyTorch training/tool code for Polygon-RNN++ (CVPR 2018)
703 stars 106 forks source link

Upgrading to Pytorch 1.3/Python3 #35

Open bearpelican opened 4 years ago

bearpelican commented 4 years ago

Hi, I'm trying to update the code to use python 3 and pytorch 1.3.

When running inference, I'm getting a different output than python 2 and pytorch 0.4.

Do you know what needs to change? SkipResnet output matches when I set Upsample(align_corners=False) PolyGGNN has a different output

LetsGoFir commented 4 years ago

How about your performance with torch1.3?