昇思25天學習打卡營第7天|保存與加載

Python語言 AI框架:Mindspore

1.模型構建

class Network(nn.Cell):def __init__(self):super().__init__()self.flatten = nn.Flatten()self.dense_relu_sequential = nn.SequentialCell(nn.Dense(28*28, 512, weight_init="normal", bias_init="zeros"),nn.ReLU(),nn.Dense(512, 512, weight_init="normal", bias_init="zeros"),nn.ReLU(),nn.Dense(512, 10, weight_init="normal", bias_init="zeros"))def construct(self, x):x = self.flatten(x)logits = self.dense_relu_sequential(x)return logits
model = Network()
print(model)

在這里插入圖片描述


2.模型保存

mindspore.save_checkpoint(model, "model.ckpt")

3.模型導出-mindir格式

除Checkpoint外,MindSpore提供了云側(訓練)和端側(推理)統一的中間表示(Intermediate Representation,IR)。可使用export接口直接將模型保存為MindIR。

model = network()
inputs = Tensor(np.ones([1, 1, 28, 28]).astype(np.float32))
mindspore.export(model, inputs, file_name="model", file_format="MINDIR")

4.加載保存模型

要加載模型權重,需要先創建相同模型的實例,然后使用load_checkpoint和load_param_into_net方法加載參數。

model = network()
param_dict = mindspore.load_checkpoint("model.ckpt")
param_not_load, _ = mindspore.load_param_into_net(model, param_dict)
print(param_not_load)  # 正確輸出[ ]

5.加載導出模型

nn.GraphCell僅支持圖模式。

mindspore.set_context(mode=mindspore.GRAPH_MODE)graph = mindspore.load("model.mindir")
model = nn.GraphCell(graph)
outputs = model(inputs)
print(outputs.shape)

MindIR同時保存了Checkpoint和模型結構,因此需要定義輸入Tensor來獲取輸入shape。


運行結果-模型保存情況如下:
在這里插入圖片描述

本文來自互聯網用戶投稿,該文觀點僅代表作者本人,不代表本站立場。本站僅提供信息存儲空間服務,不擁有所有權,不承擔相關法律責任。
如若轉載,請注明出處:http://www.pswp.cn/diannao/40579.shtml
繁體地址,請注明出處:http://hk.pswp.cn/diannao/40579.shtml
英文地址,請注明出處:http://en.pswp.cn/diannao/40579.shtml

如若內容造成侵權/違法違規/事實不符,請聯系多彩編程網進行投訴反饋email:809451989@qq.com,一經查實,立即刪除!

相關文章

力扣雙指針算法題目:移動零

1.題目 . - 力扣&#xff08;LeetCode&#xff09; 2.思路解析 這個題目的思路和“使用遞歸排序快速排序解決數組的排序問題”相同 class solution { public:void QuickSort(vector<int>& nums, int left, int right){if (left > right) return;int key left…

VSCode常用的一些插件

Chinese (Simplified) 漢語&#xff08;簡體&#xff09;拓展包。 Auto Close Tag 可以自動增加xml/html的閉合標簽。 CodeSnap 截圖神器。截圖效果在下面。 Dracula Official vscode一個很好看的主題。 Git Graph git管理工具。 GitHub Repositories 有了它&#xff0c;不…

AI一鍵音頻轉文字工具 速度超快,支持實時轉換,無需聯網,本地整合包下載

這是 CapsWriter-Offline &#xff0c;一個 PC 端的語音輸入、字幕轉錄工具。可用實現簡單一鍵將音頻文件轉換成文字的懶人工具。 兩個功能&#xff1a; 1、實時轉換&#xff0c;按下鍵盤上的 大寫鎖定鍵&#xff0c;錄音開始&#xff0c;當松開 大寫鎖定鍵 時&#xff0c;就會…

TaskDisplayArea、ActivityStack、Task以及ActivityRecord之間的關系

目錄 一、TaskDisplayArea、ActivityStack、TaskRecord以及ActivityRecord關系圖 二、TaskDisplayArea、ActivityStack、Task以及ActivityRecord相關代碼 1、ActivityRecord 2、Task 3、ActivityStack 4、TaskDisplayArea 前言&#xff1a;對TaskDisplayArea、ActivitySt…

go語言day08 泛型 自定義錯誤處理 go:協程

泛型&#xff1a; 拋錯誤異常 實現error接口類型 用java語言解釋的話&#xff0c;實現類需要重寫error類型的抽象方法Error().這樣就可以自定義異常處理。 回到go語言&#xff0c;在Error()方法中用*argError 這樣一個指針類來充當error接口的實現類。 在f2()方法中定義返回值…

CONFIG_FEATURE_MOUNT_NFS

CONFIG_FEATURE_MOUNT_NFS是一個Linux內核配置選項&#xff0c;它允許系統支持掛載NFS文件系統。如果您在編譯內核或構建根文件系統時遇到關于NFS支持的問題&#xff0c;您可能需要檢查這個配置選項是否被啟用。在某些情況下&#xff0c;例如使用BusyBox作為工具集時&#xff0…

漏洞分析|OpenSSH漏洞(CVE-2024-6387)

一、網傳漏洞POC信息 漏洞編號&#xff1a;CVE-2024-6387 漏洞名稱&#xff1a;OpenSSH regreSSHion 漏洞 POC上傳者&#xff08;作者不確定&#xff09;&#xff1a;7etsuo 發布日期&#xff1a;2024-07-01 漏洞類型&#xff1a;遠程代碼執行&#xff08;RCE&#xff09;…

昇思MindSpore學習筆記3-03熱門LLM及其他AI應用--基于MobileNetv2的垃圾分類

摘要&#xff1a; MindSpore AI框架使用MobileNetv2模型開發垃圾分檢代碼。檢測本地圖像中的垃圾物體&#xff0c;保存檢測結果到文件。記錄了開發過程和步驟&#xff0c;包括環境準備、數據下載、加載和預處理、模型搭建、訓練、測試、推理應用等。 1、實驗目的 了解垃圾分…

解決pip默認安裝位置在C盤方法

新版python中使用pip命令將opencv庫安裝到base環境中 首先我們打開命令控制窗口&#xff0c;激活base環境&#xff0c;輸入conda activate base 然后檢查一下自己base環境中是否安裝opencv庫&#xff0c;輸入conda list 往下找&#xff0c;找到o開頭的地方&#xff0c;發現是…

達夢數據庫 頁大小與數據庫字段長度的關系

對于達夢數據庫實例而言&#xff0c;頁大小 (page_size)、簇大小 (extent_size)、大小寫敏感 (case_sensitive)、字符集 (charset) 這四個參數&#xff0c;一旦確定無法修改&#xff1b;如果過程中發現這些數據設置的不對&#xff0c;只能是重新新建數據庫實例&#xff0c;而不…

GraphQL與RESTful API的區別和優勢

GraphQL GraphQL是一種用于API設計的語言和查詢協議&#xff0c;由Facebook于2015年推出。它允許客戶端向服務器指定他們需要的數據字段&#xff0c;而不是像RESTful API那樣請求整個資源然后過濾數據。在GraphQL中&#xff0c;客戶端發送一個單一的請求&#xff0c;而服務器返…

通過npm安裝OpenLayers庫,vue3+ts環境下OpenLayers實現加載本地高德離線地圖并添加標記點、標記點氣泡及標記點氣泡按鈕事件

在Vue 3和TypeScript的環境下使用OpenLayers來加載高德地圖的離線瓦片&#xff0c;并添加標記點、標記點氣泡以及處理氣泡上的按鈕事件&#xff0c;涉及到幾個步驟。首先&#xff0c;需要明確高德地圖的瓦片數據格式和如何配置OpenLayers以使用這些瓦片。接著&#xff0c;我們需…

零信任沙箱是什么?零信任沙箱有什么作用?

零信任沙箱是什么&#xff1f;零信任沙箱有什么作用&#xff1f; 在當今數字化時代&#xff0c;數據安全已成為各行各業的核心關注點。零信任沙箱作為一種新興的安全技術&#xff0c;不僅適用于政府和金融等關鍵領域&#xff0c;其實用性覆蓋了更廣泛的場景&#xff0c;如醫療…

數智化配補調:零售品牌增長新引擎

隨著科技的不斷進步和消費者需求的日益個性化、多元化&#xff0c;傳統服裝行業正面臨著前所未有的挑戰與機遇。在這個快速變化的時代&#xff0c;如何精準把握市場脈搏&#xff0c;實現庫存的高效管理&#xff0c;成為了服裝品牌生存與發展的關鍵。數智化配補調策略應運而生&a…

mysql定時備份數據庫

文章目錄 核心目標思路具體方法一、編寫腳本二、修改文件屬性三、找一個mysqldump文件四、把.sh放到定時器里 其它&#xff1a;windows的腳本 核心目標 解決數據庫定時備份的工作。centos環境。 思路 用centos的crontab定時執行腳本。 具體方法 一、編寫腳本 編寫backup_…

精準控制:Eureka服務續約間隔配置全指南

精準控制&#xff1a;Eureka服務續約間隔配置全指南 在微服務架構中&#xff0c;服務的發現與注冊是確保服務間有效通信的關鍵。Eureka&#xff0c;作為Netflix開源的服務發現框架&#xff0c;提供了一種優雅的方式來實現服務的注冊與發現。然而&#xff0c;服務續約間隔的配置…

vue單獨部署到寶塔教程

配置反向代理 注意:如果目標網站是https則寫https否則寫http 2.關于解決部署后無法刷新,直接報錯404 location / { try_files $uri $uri/ /index.html; }

程序員魚皮的保姆級寫簡歷指南第三彈,簡歷常見問題和建議匯總

大家好&#xff0c;我是程序員魚皮。做知識分享這些年來&#xff0c;我看過太多簡歷、也幫忙修改過很多的簡歷&#xff0c;發現很多同學是完全不會寫簡歷的、會犯很多常見的問題&#xff0c;不能把自己的優勢充分展示出來&#xff0c;導致錯失了很多面試機會&#xff0c;實在是…

PostgreSQL LIMIT 子句

PostgreSQL LIMIT 子句 PostgreSQL 是一種功能強大的開源對象關系數據庫管理系統&#xff0c;廣泛用于各種應用中。在處理大量數據時&#xff0c;我們通常只需要檢索部分記錄&#xff0c;而不是整個數據集。這時&#xff0c;LIMIT 子句就變得非常有用。本文將詳細介紹 Postgre…

代碼隨想錄Day74(圖論Part10)

94. 城市間貨物運輸| &#xff08;Bellman_ford隊列優化版 / SPFA&#xff09; 題目&#xff1a;94. 城市間貨物運輸 I (kamacoder.com) 思路&#xff1a; Bellman_ford 算法 每次都是對所有邊進行松弛&#xff0c;其實是多做了一些無用功。 只需要對 上一次松弛的時候更新過的…