diff --git a/imperative/python/megengine/functional/vision.py b/imperative/python/megengine/functional/vision.py index f067f08f..b713a829 100644 --- a/imperative/python/megengine/functional/vision.py +++ b/imperative/python/megengine/functional/vision.py @@ -582,7 +582,8 @@ def interpolate( "nearest": "nearest", "bicubic": "cubic", } - + if inp.dtype == np.float16: + inp = inp.astype("float32") op = builtin.Resize(imode=mode_map[mode], format="NCHW") shape = astensor1d(dsize, inp, dtype="int32", device=inp.device) (ret,) = apply(op, inp, shape)