快捷導(dǎo)航

解決ImportError:cannot import name ‘Flatten‘ from ‘torch.nn‘問題

更新時(shí)間：2023年06月15日 09:09:58 作者：全幼兒園最聰明

這篇文章主要介紹了解決ImportError:cannot import name ‘Flatten‘ from ‘torch.nn‘問題，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。如有錯(cuò)誤或未考慮完全的地方，望不吝賜教

問題描述

在代碼中從torch.nn引入Flatten報(bào)錯(cuò)：

解決方案

進(jìn)入python交互式環(huán)境

輸入以下代碼，查看pytorch版本：

import torch
print(torch.__version__)

可以看到pytorch版本是1.1.0

版本太低不包含Flatten屬性

因此，一個(gè)辦法就是升級(jí)pytorch版本，但我們知道下載安裝pytorch并不容易，若沒有鏡像下載緩慢且需要考慮顯卡驅(qū)動(dòng)、conda、torch三者的版本關(guān)系，升級(jí)可能導(dǎo)致版本不匹配conda無(wú)法使用gpu等問題。

另一個(gè)簡(jiǎn)單的辦法是自己寫Flatten類，這里我需要的功能是僅是扁平化，

參考代碼如下：

class Flatten(nn.Module):
    def forward(self, input):
        return input.view(input.size(0), -1)

引入后重新運(yùn)行代碼，沒有報(bào)錯(cuò)

總結(jié)

以上為個(gè)人經(jīng)驗(yàn)，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。

您可能感興趣的文章:

相關(guān)文章

python requests post的使用方式
這篇文章主要介紹了python requests post的使用方式，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。如有錯(cuò)誤或未考慮完全的地方，望不吝賜教
2023-03-03
python中g(shù)eopandas庫(kù)安裝出現(xiàn)各種問題的解決辦法
這篇文章主要介紹了關(guān)于python中g(shù)eopandas庫(kù)安裝出現(xiàn)各種問題的解決辦法,總結(jié)了在Windows下兩種安裝geopandas庫(kù)的方法,方法一是在新環(huán)境下使用conda命令安裝,方法二通過離線安裝GDAL、Fiona、Pyproj、Rtree、Shapely五個(gè)庫(kù),再用pip安裝geopandas,需要的朋友可以參考下
2024-11-11
基于Python實(shí)現(xiàn)成語(yǔ)填空游戲的示例代碼
成語(yǔ)填空想必大家都是十分熟悉的了，特別是有在上小學(xué)的家長(zhǎng)肯定都有十分深刻的印象。當(dāng)然了你也別小看了成語(yǔ)調(diào)控小游戲，有的時(shí)候知識(shí)儲(chǔ)備不夠，你還真的不一定猜得出來(lái)是什么。本文就來(lái)用Python編寫一個(gè)簡(jiǎn)單的成語(yǔ)填空游戲，感興趣的可以了解下
2023-02-02
python 切片和range()用法說明
首先需要明白，可迭代對(duì)象，按照正數(shù)索引（正序）是從0開始的，按照負(fù)數(shù)索引（逆序）是從－1開始的
2013-03-03
Python控制臺(tái)輸出俄羅斯方塊的方法實(shí)例
這篇文章主要給大家介紹了關(guān)于Python控制臺(tái)輸出俄羅斯方塊的相關(guān)資料,文中通過示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧
2021-04-04
解決django接口無(wú)法通過ip進(jìn)行訪問的問題
這篇文章主要介紹了解決django接口無(wú)法通過ip進(jìn)行訪問的問題，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過來(lái)看看吧
2020-03-03
python?Tkinter實(shí)例詳解
tkinter(Tk?interface)是Python的標(biāo)準(zhǔn)GUl庫(kù)，支持跨平臺(tái)的GUl程序開發(fā)。tkinter適合小型的GUl程序編寫，也特別適合初學(xué)者學(xué)習(xí)GUl編程，這篇文章主要介紹了python?Tkinter詳解,需要的朋友可以參考下
2023-03-03
如何使用python編寫一個(gè)簡(jiǎn)單的課時(shí)記錄系統(tǒng)
編寫一個(gè)應(yīng)用系統(tǒng)需要多方面的知識(shí)和技能,下面這篇文章主要給大家介紹了關(guān)于如何使用python編寫一個(gè)簡(jiǎn)單的課時(shí)記錄系統(tǒng)的相關(guān)資料,文中通過代碼介紹的非常詳細(xì),對(duì)大家學(xué)習(xí)或者使用python具有一定的參考借鑒價(jià)值,需要的朋友可以參考下
2024-04-04
如何使用yolov5輸出檢測(cè)到的目標(biāo)坐標(biāo)信息
YOLOv5是一系列在 COCO 數(shù)據(jù)集上預(yù)訓(xùn)練的對(duì)象檢測(cè)架構(gòu)和模型,下面這篇文章主要給大家介紹了關(guān)于如何使用yolov5輸出檢測(cè)到的目標(biāo)坐標(biāo)信息的相關(guān)資料,需要的朋友可以參考下
2022-03-03
Python字符串詳細(xì)介紹
這篇文章主要介紹了Python字符串詳解,本文講解了字符串相關(guān)知識(shí)、字符串的一些特性、原始字符串、unicode字符串、字符串的常用操作方法、內(nèi)建函數(shù)列表等內(nèi)容,需要的朋友可以參考下
2015-05-05