pytorch 代码块积累

最新推荐文章于 2024-07-22 14:33:33 发布

犟小孩

最新推荐文章于 2024-07-22 14:33:33 发布

阅读量409

点赞数 8

分类专栏：技术文档编程相关代码查询文章标签： pytorch 人工智能 python

本文链接：https://blog.csdn.net/qq_55542491/article/details/134779285

版权

技术文档同时被 3 个专栏收录

21 篇文章 0 订阅

订阅专栏

代码查询

16 篇文章 0 订阅

订阅专栏

编程相关

11 篇文章 0 订阅

订阅专栏

模型打印每层shape

import torch

from models.enet import  ENet
from torchsummary import summary #pip install torchsummary
model=ENet(12)
aaa=torch.load("./save/ENet",map_location="cpu")
# print(aaa)
model.load_state_dict(aaa["state_dict"])

summary(model, (3, 256, 256))