pytorch 代码块积累

模型打印每层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))
相关推荐
sjr20013 分钟前
从huggingface下载模型时有哪些文件?
人工智能·机器学习
moz与京3 分钟前
【面试向】热门技术话题(上)
人工智能·物联网·机器学习·面试·web3·区块链·元宇宙
wyfwyf___21 分钟前
5G+IoT+AI:新质工业新图景,从预测性维护到全链路数智化
人工智能·科技·物联网·5g·信息与通信
gptplusplus1 小时前
AI智能体(Agent):从“辅助决策”到“自主行动”,重新定义下一个商业时代
人工智能
别忘了微笑啊1 小时前
hCaptcha 图像识别 API 对接说明
人工智能
尝试经历体验1 小时前
pycharm突然不能正常运行
python·深度学习·pycharm
大千AI助手1 小时前
灾难性遗忘:神经网络持续学习的核心挑战与解决方案
人工智能·深度学习·神经网络·大模型·llm·持续学习·灾难性遗忘
gotouniverse1 小时前
之前自学RAG时做的调研
人工智能
jay神1 小时前
基于Python的商品爬取与可视化系统
爬虫·python·数据分析·毕业设计·可视化系统
新智元1 小时前
刚刚,英伟达祭出下一代 GPU!狂飙百万 token 巨兽,投 1 亿爆赚 50 亿
人工智能·openai