- 博客(6)
- 收藏
- 关注
原创 如何冻结 YOLO参数进行训练
在训练yolo模型时,有时需要冻结部分模型参数来保留预训练模型的知识,避免某些层参数更新对最终模型性能的负面影响。本文将以 YOLOv8 为例,介绍如何冻结模型部分参数并进行训练。上述模型由 22 层具有可训练参数的模块组成。了解模型结构是冻结参数的前提。要冻结特定参数,首先需要知道参数名称。通过查看输出,可以确定要冻结的层对应的参数名称。
2025-01-05 14:36:45
1246
原创 Diffusion Model原理及Stable Diffusion
Stable Diffusion由三部分组成:CLIPTextEncoder、diffusion model、以及VAE Decoder,要想弄懂Stable Diffusion,必须懂这三个部分。
2024-05-16 21:47:56
806
1
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人