ModelArts模型管理導(dǎo)入FrozenGraph格式模型

當(dāng)前ModelArts文檔中只有SavedModel格式的案例

ModelArts模型導(dǎo)入事實(shí)上也支持使用FrozenGraph進(jìn)行模型導(dǎo)入。

通過(guò)閱讀文檔https://support.huaweicloud.com/engineers-modelarts/modelarts_23_0093.html

從文檔中可以看到有如下方法進(jìn)行重寫(xiě)

那么這兩種模型導(dǎo)入的差別在哪里呢。

如果導(dǎo)入的模型文件是SavedModel，那么一般情況下是不需要 實(shí)現(xiàn)__init__(self,model_name,model_path)方法。

但是如果導(dǎo)入的模型是FrozenGraph就需要實(shí)現(xiàn)__init__(self,model_name,model_path)方法，并且在此方法中實(shí)現(xiàn)模型的加載。

以tensorflow2.0為例

https://support.huaweicloud.com/engineers-modelarts/modelarts_23_0301.html

__init__(self,model_name,model_path)方法中調(diào)用了load_model，因此只要在load_model()方法中實(shí)現(xiàn)加載FozenGraph格式的模型就可以了。

AI開(kāi)發(fā)平臺(tái)ModelArts

版權(quán)聲明：本文內(nèi)容由網(wǎng)絡(luò)用戶投稿，版權(quán)歸原作者所有，本站不擁有其著作權(quán)，亦不承擔(dān)相應(yīng)法律責(zé)任。如果您發(fā)現(xiàn)本站中有涉嫌抄襲或描述失實(shí)的內(nèi)容，請(qǐng)聯(lián)系我們jiasou666@gmail.com 處理，核實(shí)后本網(wǎng)站將在24小時(shí)內(nèi)刪除侵權(quán)內(nèi)容。