运行darts 出现错误

RuntimeError: cuda runtime error (11) : invalid argument at /pytorch/aten/src/THC/THCGeneral.cpp:383

查到的解决方法：

cudnn.benchmark = False

IndexError: invalid index of a 0-dim tensor. Use `tensor.item()` in Python or `tensor.item()` in C++ to convert a 0-dim tensor to a number

解决方法：

https://blog.csdn.net/smiling614/article/details/89469044