Browse Source

fix(mge/function): fix interpolate unsupport fp16 error

GitOrigin-RevId: 7fc6271986
tags/v1.6.0-rc1
Megvii Engine Team 3 years ago
parent
commit
29d24dbb80
1 changed files with 2 additions and 1 deletions
  1. +2
    -1
      imperative/python/megengine/functional/vision.py

+ 2
- 1
imperative/python/megengine/functional/vision.py View File

@@ -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)


Loading…
Cancel
Save