RuntimeError: isDifferentiableType(variable.scalar_type()) INTERNAL ASSERT FAILED at "/opt/conda/conda-bld/pytorch_1595629401553/work/torch/csrc/autograd/functions/utils.h":59, please report a bug to PyTorch.
代码加上:
with torch.no_grad():