报错如下：RuntimeError: kind_.is_prim() INTERNAL ASSERT FAILED at /pytorch/torch/csrc/jit/ir.cpp:904, please report a bug to PyTorch. Only prim ops are allowed to not have a registered operator but aten::_convolution doesn't have one either. We don't know if this op has side effects. (hasSideEffects at /pytorch/torch/csrc/jit/ir.cpp:904)
报错原因：
Pytorch使用tensorboardX可视化，使用语句from tensorboardX import SummaryWriter
解决方法：
将from tensorboardX import SummaryWriter
改成from torch.utils.tensorboard import SummaryWriter