Fix the UpsamplingNearest2d's symbolic

houseroad · houseroad · commit 3d15ef34ee7e · 2017-11-29T12:21:18.000-08:00
diff --git a/torch/nn/_functions/thnn/upsampling.py b/torch/nn/_functions/thnn/upsampling.py
@@ -175,8 +175,10 @@ def symbolic(g, input, size=None, scale_factor=None):
         if scale_factor is None:
             scale_factor = 1.0
         if size is not None and set(size) != set([None]):
-            warnings.warn("ONNX export failed on UpsamplingNearest2d because size is not supported")
-        return g.op("ResizeNearest", input, width_scale_f=scale_factor, height_scale_f=scale_factor)
+            from torch.onnx.symbolic import _unimplemented
+            return _unimplemented("UpsamplingNearest2d", "size")
+        return g.op("Upsample", input, width_scale_f=scale_factor,
+                    height_scale_f=scale_factor, mode_s="nearest")
 
     @staticmethod
     def forward(ctx, input, size=None, scale_factor=None):
