Commit 1b06e3e0 by 前钰

Add new file

parent f97f7576
# 模型训练策略优化
# 模型训练策略优化
网盘链接包含ResNet18 权重文件,课程内项目见图像分类章节
## 网盘下载链接
* 链接: [百度网盘](https://pan.baidu.com/s/16WHxegUqrmxOauchwPmqRg)
* 提取码: `8888`
## 文件说明
| 文件名 | 功能说明 |
| -------------- | -------------------------------------------------- |
| `huitu.ipynb` | 根据训练日志绘制训练过程中的 loss 和 accuracy 曲线,便于观察模型收敛情况 |
| `train_es.py` | 包含训练脚本,实现了早停(Early Stopping)和加权损失函数,用于提升模型训练稳定性与精度 |
| `学习率调度器/` | 文件夹中包含多种深度学习学习率调度器示例,可在训练中灵活选择以优化收敛速度与性能 |
| `resnet18权重文件` | 预训练 ResNet18 模型权重,可直接加载用于图像分类任务 |
## 特性
* 支持早停机制,防止过拟合。
* 支持加权损失函数,提高少数类别识别能力。
* 提供多种学习率调度器,灵活优化训练策略。
* 可直接加载 ResNet18 权重文件,减少训练时间。
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment