1.repeat
PyTorch中Tensor的复制tensor.repeat()和repeat_interleave()区别_Kevinllli的博客-CSDN博客_tensor复制
[Pytorch函数].repeat()_Jeanshoe的博客-CSDN博客_pytorch repeat函数
2.train() eval() no_grad()
关于Pytorch中的train()和eval()(以及no_grad())_pytorch eval函数_我是一个对称矩阵的博客-CSDN博客
2.报错block: [0,0,0], thread: [16,0,0] Assertion t 0 && t n_classes failed.
CUDA error: CUBLAS_STATUS_NOT_INITIALIZED。。。。。。一大串
造成这个问题的原因就是在做分类任务时,训练数据中存在超出分类数目的标签。比如我一共设置了8个类,但是训练数据中的标签里出现了9,就会报这个错误。
在pytorch里小于0的类别标签也会报这个错。