ggml教程|mnist手写体识别量化推理
ggml教程|mnist手写体识别量化推理 MNIST手写体识别是经典的机器学习问题,可以被称作机器学习的hello world了,我希望通过mnist来作为系列教程的第一节,来介绍如何使用ggml量化,推理一个模型。这个教程将会使用pytorch来训练一个简单的全连接神经网络,然后使用ggml量化,最后使用ggml推理这个模型。 代码开源在仓库ggml-tutorial 训练模型 首先我们使用pytorch来训练一个简单的全连接神经网络,代码在train.py 文件中,训练...
ControlNet训练自己数据集
2024.1.20更新 controlnet发布快一年了,diffusers已经有了很完整的生态,建议直接使用第二种方式diffusers进行训练+推理 从官方仓库训练 官方教程 https://github.com/lllyasviel/ControlNet/blob/main/docs/train.md 环境配置 先看一下有没有显卡 nvidia-smi 首先下载整个仓库 git clone https://github.com/lllyasviel/ControlNet.git 然后创建conda虚拟环境(选做...