【Pytorch】解决使用BucketIterator.splits警告volatile was removed and now has no effect. Use `with torch.no_g
PyTorch 解决 and with No use has was
2023-09-14 09:12:40 时间
问题
使用data.BucketIterator.splits去封装训练集和验证集,在遍历的时候出现警告UserWarning: volatile was removed and now has no effect. Use with torch.no_grad():
instead. return Variable(arr, volatile=not train)
from torchtext import data
train_iter,valid_iter= data.BucketIterator.splits((train_data,valid_data),
batch_size=batch_size,
sort_key=lambda x: len(x.text),
repeat=False,
shuffle=True)
with torch.no_grad():
for idx, batch in enumerate(val_iter):# 在一行出现警告
pass
解决
分开封装,不用data.BucketIterator.splits而是用data.BucketIterator
from torchtext import data
train_iter = data.BucketIterator((train_data), batch_size=batch_size,
sort_key=lambda x: len(x.text),
repeat=False, shuffle=True)
valid_iter = data.BucketIterator((valid_data),
batch_size=batch_size,
sort_key=lambda x: len(x.text),
repeat=False,
shuffle=True)
相关文章
- 从零开始安装pytorch,并在pycharm中使用
- python2.7安装pytorch_PyTorch安装「建议收藏」
- PyTorch—torchvision.models导入预训练模型—残差网络代码讲解
- PyTorch和Meta「分家」!以后归Linux基金会管了
- Ubuntu20.04安装cuda cudnn pytorch pycharm记录
- PyTorch实现的ResNet50、ResNet101和ResNet152
- 使用PyTorch进行知识蒸馏的代码示例
- PyTorch-24h 03_图像分类
- Github1.3万星,迅猛发展的JAX对比TensorFlow、PyTorch
- 基于PyTorch,集合17种方法,南京大学等提出小样本算法库LibFewShot