Closed yyq closed 1 year ago
推理量化模型的时候会遇到报错：
RuntimeError: Inplace update to inference tensor outside InferenceMode is not allowed.You can make a clone to get a normal tensor before doing inplace update.See https://github.com/pytorch/rfcs/pull/17 for more details.
新增一行代码：
with torch.inference_mode(mode=True):
推理量化模型的时候会遇到报错：
新增一行代码：