欧美bbbwbbbw肥妇,免费乱码人妻系列日韩,一级黄片

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

 更新時間：2023年06月15日 09:09:58   作者：全幼兒園最聰明  
這篇文章主要介紹了解決ImportError:cannot import name ‘Flatten‘ from ‘torch.nn‘問題，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教

問題描述

在代碼中從torch.nn引入Flatten報錯：

解決方案

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

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

import torch
print(torch.__version__)

可以看到pytorch版本是1.1.0

版本太低不包含Flatten屬性

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

另一個簡單的辦法是自己寫Flatten類，這里我需要的功能是僅是扁平化，

參考代碼如下：

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

引入后重新運(yùn)行代碼，沒有報錯

總結(jié)

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

相關(guān)文章

  • python requests post的使用方式

    python requests post的使用方式

    這篇文章主要介紹了python requests post的使用方式，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教
    2023-03-03
  • python中g(shù)eopandas庫安裝出現(xiàn)各種問題的解決辦法

    python中g(shù)eopandas庫安裝出現(xiàn)各種問題的解決辦法

    這篇文章主要介紹了關(guān)于python中g(shù)eopandas庫安裝出現(xiàn)各種問題的解決辦法,總結(jié)了在Windows下兩種安裝geopandas庫的方法,方法一是在新環(huán)境下使用conda命令安裝,方法二通過離線安裝GDAL、Fiona、Pyproj、Rtree、Shapely五個庫,再用pip安裝geopandas,需要的朋友可以參考下
    2024-11-11
  • 基于Python實(shí)現(xiàn)成語填空游戲的示例代碼

    基于Python實(shí)現(xiàn)成語填空游戲的示例代碼

    成語填空想必大家都是十分熟悉的了，特別是有在上小學(xué)的家長肯定都有十分深刻的印象。當(dāng)然了你也別小看了成語調(diào)控小游戲，有的時候知識儲備不夠，你還真的不一定猜得出來是什么。本文就來用Python編寫一個簡單的成語填空游戲，感興趣的可以了解下
    2023-02-02
  • python 切片和range()用法說明

    python 切片和range()用法說明

    首先需要明白，可迭代對象，按照正數(shù)索引（正序）是從0開始的，按照負(fù)數(shù)索引（逆序）是從－1開始的
    2013-03-03
  • Python控制臺輸出俄羅斯方塊的方法實(shí)例

    Python控制臺輸出俄羅斯方塊的方法實(shí)例

    這篇文章主要給大家介紹了關(guān)于Python控制臺輸出俄羅斯方塊的相關(guān)資料,文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧
    2021-04-04
  • 解決django接口無法通過ip進(jìn)行訪問的問題

    解決django接口無法通過ip進(jìn)行訪問的問題

    這篇文章主要介紹了解決django接口無法通過ip進(jìn)行訪問的問題，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
    2020-03-03
  • python?Tkinter實(shí)例詳解

    python?Tkinter實(shí)例詳解

    tkinter(Tk?interface)是Python的標(biāo)準(zhǔn)GUl庫，支持跨平臺的GUl程序開發(fā)。tkinter適合小型的GUl程序編寫，也特別適合初學(xué)者學(xué)習(xí)GUl編程，這篇文章主要介紹了python?Tkinter詳解,需要的朋友可以參考下
    2023-03-03
  • 如何使用python編寫一個簡單的課時記錄系統(tǒng)

    如何使用python編寫一個簡單的課時記錄系統(tǒng)

    編寫一個應(yīng)用系統(tǒng)需要多方面的知識和技能,下面這篇文章主要給大家介紹了關(guān)于如何使用python編寫一個簡單的課時記錄系統(tǒng)的相關(guān)資料,文中通過代碼介紹的非常詳細(xì),對大家學(xué)習(xí)或者使用python具有一定的參考借鑒價值,需要的朋友可以參考下
    2024-04-04
  • 如何使用yolov5輸出檢測到的目標(biāo)坐標(biāo)信息

    如何使用yolov5輸出檢測到的目標(biāo)坐標(biāo)信息

    YOLOv5是一系列在 COCO 數(shù)據(jù)集上預(yù)訓(xùn)練的對象檢測架構(gòu)和模型,下面這篇文章主要給大家介紹了關(guān)于如何使用yolov5輸出檢測到的目標(biāo)坐標(biāo)信息的相關(guān)資料,需要的朋友可以參考下
    2022-03-03
  • Python字符串詳細(xì)介紹

    Python字符串詳細(xì)介紹

    這篇文章主要介紹了Python字符串詳解,本文講解了字符串相關(guān)知識、字符串的一些特性、原始字符串、unicode字符串、字符串的常用操作方法、內(nèi)建函數(shù)列表等內(nèi)容,需要的朋友可以參考下
    2015-05-05

最新評論