RuntimeError: kind_.is_prim() INTERNAL ASSERT FAILED

RuntimeError: kind_.is_prim() INTERNAL ASSERT FAILED

tensorboardX’s problem

How to solve:

step1.

install tensorboard

step2.

change

from tensorboardX import SummaryWriter

to

from torch.utils.tensorboard import SummaryWriter

ref

你可能感兴趣的:(python,机器学习,深度学习,自然语言处理)