# fine-tuning-examples

**Repository Path**: gitee-ai/fine-tuning-examples

## Basic Information

- **Project Name**: fine-tuning-examples
- **Description**: No description available
- **Primary Language**: Unknown
- **License**: Not specified
- **Default Branch**: main
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 1
- **Forks**: 0
- **Created**: 2023-10-20
- **Last Updated**: 2024-01-15

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

# 在本地进行简单的业务模型微调

## 准备


1. 安装依赖

```bash
pip3 install torch datasets transformers
```

2. 下载数据集

```bash
git clone git@gitee.com:gitee-ai/gitee-task-owner-classification-large.git
```

## 训练

```bash
HF_ENDPOINT=https://ai.gitee.com/huggingface HF_HOME=~/.cache/gitee-ai python3 train.py
```

## 推理

```bash
HF_ENDPOINT=https://ai.gitee.com/huggingface HF_HOME=~/.cache/gitee-ai python3 run.py
```

## CPU/GPU 负载

- macOS

```bash
sudo asitop --show_cores 1  
```

- Linux

```bash
nvitop
```