torch‘ has no attribute ‘inference_ mode‘

stay overstack flow Someone said it was because of the version problem ,（ This order is in torch.1.9 above ）, But someone below said that this was not good ;

Solution ：

Replace code , take torch.inference_mode() Replace with torch.no_grad()

@torch.no_grad()
# @torch.inference_mode()

    with torch.no_grad():
    # with torch.inference_mode():