打卡第36天:模型可视化以及推理

知识点回顾: 1.三种不同的模型可视化方法:推荐torchinfo打印summary+权重分布可视化

2.进度条功能:手动和自动写法,让打印结果更加美观

3.推理的写法:评估模式

作业:调整模型定义时的超参数,对比下效果。

1.只调整隐藏层参数

python 复制代码
    def __init__(self):  
        super(MLP, self).__init__()  
        self.fc1 = nn.Linear(4, 20)  # 输入层到隐藏层  
        self.relu = nn.ReLU()  
        self.fc2 = nn.Linear(20, 3)  # 隐藏层到输出层

python运行损失并没有下降

2.调整优化器

python 复制代码
optimizer = optim.Adam(model.parameters(), lr=0.001)

3.优化器+隐藏层

@浙大疏锦行

相关推荐
IT莫染几秒前
用脚本解放双手!我写了个WorkBuddy自动签到工具
python
d1z8882 分钟前
(十八)32天GPU测试从入门到精通-TensorRT-LLM 部署与优化day16
人工智能·python·深度学习·gpu·tensorrt
txzrxz4 分钟前
c++深度搜索讲解及例题
开发语言·c++·深度搜索·例题讲解
yu85939584 分钟前
时延估计的互相关算法(MATLAB实现)
开发语言·算法·matlab
ou.cs5 分钟前
c# SemaphoreSlim保姆级教程
开发语言·网络·c#
qq_283720055 分钟前
Python 面向对象编程(OOP)从入门到精通
python·oop·面对对象
|_⊙7 分钟前
红黑树 (C++)
开发语言·c++·学习
linux_map8 分钟前
大模型微调实战指南
人工智能·python·ai·策略模式
Fate_I_C18 分钟前
Kotlin 内部类和嵌套类
java·开发语言·kotlin
昵称暂无126 分钟前
低代码平台深度测评:OutSystems vs Mendix谁更胜一筹
开发语言·低代码