pytorch数据预处理错误的解决
出错：
Traceback(mostrecentcalllast): File"train.py",line305,intrain_model(model_conv,criterion,optimizer_conv,exp_lr_scheduler) File"train.py",line145,intrain_model forinputs,age_labels,gender_labelsindataloaders[phase]: File"/home/home/anaconda3/lib/python3.6/site-packages/torch/utils/data/dataloader.py",line637,in__next__ returnself._process_next_batch(batch) File"/home/home/anaconda3/lib/python3.6/site-packages/torch/utils/data/dataloader.py",line658,in_process_next_batch raisebatch.exc_type(batch.exc_msg) RuntimeError:Traceback(mostrecentcalllast): File"/home/home/anaconda3/lib/python3.6/site-packages/torch/utils/data/dataloader.py",line138,in_worker_loop samples=collate_fn([dataset[i]foriinbatch_indices]) File"/home/home/anaconda3/lib/python3.6/site-packages/torch/utils/data/dataloader.py",line232,indefault_collate return[default_collate(samples)forsamplesintransposed] File"/home/home/anaconda3/lib/python3.6/site-packages/torch/utils/data/dataloader.py",line232,in return[default_collate(samples)forsamplesintransposed] File"/home/home/anaconda3/lib/python3.6/site-packages/torch/utils/data/dataloader.py",line209,indefault_collate returntorch.stack(batch,0,out=out) RuntimeError:invalidargument0:Sizesoftensorsmustmatchexceptindimension0.Got224and228indimension3at/pytorch/aten/src/TH/generic/THTensorMoreMath.cpp:1307
这是因为输入的大小不匹配，跟数据集有关，也跟数据预处理中的函数相关：
transforms.Resize(input_size)
该函数是按比例缩放，可能是因为该数据集的分辨率不同，所以出来的结果不是(224,224)的，解决办法是改为使用：
transforms.Resize((input_size,input_size))
即可
以上这篇pytorch数据预处理错误的解决就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持毛票票。
声明：本文内容来源于网络，版权归原作者所有，内容由互联网用户自发贡献自行上传，本网站不拥有所有权，未作人工编辑处理，也不承担相关法律责任。如果您发现有涉嫌版权的内容，欢迎发送邮件至：czq8825#qq.com（发邮件时，请将#更换为@）进行举报，并提供相关证据，一经查实，本站将立刻删除涉嫌侵权内容。