pytorch數(shù)據(jù)預(yù)處理錯誤的解決
更新時間：2020年02月20日 10:45:07 作者：慢行厚積
今天小編就為大家分享一篇pytorch數(shù)據(jù)預(yù)處理錯誤的解決，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
出錯：
Traceback (most recent call last): File "train.py", line 305, in <module> train_model(model_conv, criterion, optimizer_conv, exp_lr_scheduler) File "train.py", line 145, in train_model for inputs, age_labels, gender_labels in dataloaders[phase]: File "/home/home/anaconda3/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 637, in __next__ return self._process_next_batch(batch) File "/home/home/anaconda3/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 658, in _process_next_batch raise batch.exc_type(batch.exc_msg) RuntimeError: Traceback (most recent call last): File "/home/home/anaconda3/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 138, in _worker_loop samples = collate_fn([dataset[i] for i in batch_indices]) File "/home/home/anaconda3/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 232, in default_collate return [default_collate(samples) for samples in transposed] File "/home/home/anaconda3/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 232, in <listcomp> return [default_collate(samples) for samples in transposed] File "/home/home/anaconda3/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 209, in default_collate return torch.stack(batch, 0, out=out) RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got 224 and 228 in dimension 3 at /pytorch/aten/src/TH/generic/THTensorMoreMath.cpp:1307
這是因為輸入的大小不匹配，跟數(shù)據(jù)集有關(guān)，也跟數(shù)據(jù)預(yù)處理中的函數(shù)相關(guān)：
transforms.Resize(input_size)
該函數(shù)是按比例縮放，可能是因為該數(shù)據(jù)集的分辨率不同，所以出來的結(jié)果不是(224,224)的，解決辦法是改為使用：
transforms.Resize((input_size, input_size))
即可
以上這篇pytorch數(shù)據(jù)預(yù)處理錯誤的解決就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關(guān)文章
本文將向讀者展示如何利用Python編寫自動化腳本,以檢查磁盤空間使用情況,無論你是經(jīng)驗豐富的系統(tǒng)管理員,還是對Python自動化充滿興趣的開發(fā)者,本文都將為你提供實用的腳本示例和詳細(xì)的解析步驟,幫助你快速掌握磁盤空間監(jiān)控的自動化方法,需要的朋友可以參考下2024-08-08
策略模式屬于Python-設(shè)計模式中行為模式之一，該模式定義了一系列算法，并將每個算法封裝起來，使它們可以相互替換。本文將通過示例詳細(xì)講解這一模式，需要的可以參考一下2023-02-02
這篇文章主要介紹了幾款Python編譯器比較與推薦(小結(jié))，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2020-10-10
pycharm是一種Python IDE，能夠幫助我們在編寫代碼時提高效率。 這篇文章主要介紹了Pycharm小白級簡單使用教程,需要的朋友可以參考下2020-01-01
裝飾器的使用是Python的高級技巧之一,通過Decorator裝飾器能夠帶來函數(shù)式編程中所需的很多特性,這里我們就來跟隨文章一同深入理解Python中裝飾器的用法,需要的朋友可以參考下2016-06-06
python修改linux中文件(文件夾)的權(quán)限屬性操作
這篇文章主要介紹了python修改linux中文件(文件夾)的權(quán)限屬性操作，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2020-03-03
這篇文章主要介紹了解決PyQt5 無邊框后窗口的移動問題，具有很好的參考價值，希望對大家有所幫助，一起跟隨小編過來看看吧2021-03-03
Windows和夜神模擬器上抓包程序mitmproxy的安裝使用詳解
mitmproxy是一個支持HTTP和HTTPS的抓包程序，有類似Fiddler、Charles的功能，只不過它是一個控制臺的形式操作，這篇文章主要介紹了Windows和夜神模擬器上抓包程序mitmproxy的安裝使用詳解,需要的朋友可以參考下2022-10-10
python3的一個天坑問題及解決方法:報錯UnicodeDecodeError: ‘utf-8‘
在調(diào)試程序發(fā)現(xiàn)python3的一個天坑問題:報錯UnicodeDecodeError: ‘utf-8‘ codec can‘t decode byte 0xa3 in position 59: invalid,特此曝光,為眾位開發(fā)朋友提個醒2023-09-09
python實現(xiàn)在函數(shù)圖像上添加文字和標(biāo)注的方法
今天小編就為大家分享一篇python實現(xiàn)在函數(shù)圖像上添加文字和標(biāo)注的方法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2019-07-07