Pytorch 小问题

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的类别标签也会报这个错。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值