# mm-train

**Repository Path**: mr-abc/mm-train

## Basic Information

- **Project Name**: mm-train
- **Description**: train models using MMEngine
- **Primary Language**: Unknown
- **License**: MIT
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 0
- **Forks**: 0
- **Created**: 2023-09-04
- **Last Updated**: 2023-09-04

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

## [MMEngine](https://mmengine.readthedocs.io/zh_CN/latest/get_started/introduction.html)
The examples come from MMEngine.

### [Getting Started with MMEngine](https://mmengine.readthedocs.io/zh_CN/latest/get_started/15_minutes.html)
- 01_resnet_train.py: train ResNet50 on CIFAR10

```
python 01_resnet_train.py
python 01_resnet_train.py --data_dir data/cifar10 --workdir runs/resnet --epochs 5
```

### [GAN](https://mmengine.readthedocs.io/zh_CN/latest/examples/train_a_gan.html)
- 02_gan_train.py: train a GAN on MNIST
- 02_gan_test.py: generate an image use the GAN

```
python 02_gan_train.py
python 02_gan_train.py --data_dir data --workdir runs/gan --epochs 50
python 02_gan_test.py ----checkpoint runs/gan/epoch_50.pth --result runs/gan/result.png
```

### [Segmentation](https://mmengine.readthedocs.io/zh_CN/latest/examples/train_seg.html)
- 03_camvid_train.py: train a segmentator on CamVid

```
python 03_camvid_train.py
python 03_camvid_train.py --data_dir data/CamVid --workdir runs/camvid --epochs 10
```