神经网络
天莓烦恼
这个作者很懒,什么都没留下…
展开
-
RoBERTa 阅读笔记
总的来说,bert在模型上并没有做太多的改进,只是在训练数据和训练策略上进行了一个改进。原创 2023-04-22 11:20:10 · 228 阅读 · 0 评论 -
Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu 解决
如果是因为调用了torchmetrics的评价指标如Accuracy,F1等的话,那么就要注意是不是因为这些指标和数据不在同一块设备上。如果你找到的不是因为data或者label没有加上。原创 2022-10-23 00:04:15 · 756 阅读 · 0 评论