diff --git a/python/tvm/relay/frontend/pytorch.py b/python/tvm/relay/frontend/pytorch.py index 32a7d9c5d8b9..b02e59b2654b 100644 --- a/python/tvm/relay/frontend/pytorch.py +++ b/python/tvm/relay/frontend/pytorch.py @@ -2756,7 +2756,7 @@ def interpolate(self, inputs, input_types): for i in [0, 1]: size, _ = try_infer_value( inputs[1][i], - lambda ret: ret.astype(np.int), + lambda ret: ret.astype(int), lambda: _op.expand_dims(inputs[1][i], axis=0), ) out_size.append(size)