mirrors/fast-reid
1
0
Fork 0
mirror of https://github.com/JDAI-CV/fast-reid.git synced 2025-06-03 14:50:47 +08:00
Code Issues Projects Releases Wiki Activity
fast-reid/projects/FastCls
History
liaoxingyu e26182e6ec make lr warmup by iter 
Summary: change warmup way by iter not by epoch, which will make it more flexible when training small epochs
2021-01-22 11:17:21 +08:00
..
configs
make lr warmup by iter
2021-01-22 11:17:21 +08:00
fastcls
update fastreid V1.0
2021-01-18 11:36:38 +08:00
README.md
update fastreid V1.0
2021-01-18 11:36:38 +08:00
train_net.py
update fastreid V1.0
2021-01-18 11:36:38 +08:00

README.md

FastCls in FastReID

This project provides a baseline and example for image classification based on fastreid.

Datasets Preparation

We refer to pytorch tutorial for dataset preparation. This is just an example for building a classification task based on fastreid. You can customize your own datasets and model.

Usage

If you want to train models with 4 gpus, you can run

python3 projects/FastCls/train_net.py --config-file projects/FastCls/config/base-cls.yml --num-gpus 4