mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-09-23 10:30:03 +00:00
[example] add train resnet/vit with booster example (#3694)
* [example] add train vit with booster example * [example] update readme * [example] add train resnet with booster example * [example] enable ci * [example] enable ci * [example] add requirements * [hotfix] fix analyzer init * [example] update requirements
This commit is contained in:
@@ -10,6 +10,12 @@ This example provides a training script, which provides an example of finetuning
|
||||
- `--target_f1`: Target f1 score. Raise exception if not reached. Defaults to `None`.
|
||||
|
||||
|
||||
### Install requirements
|
||||
|
||||
```bash
|
||||
pip install -r requirements.txt
|
||||
```
|
||||
|
||||
### Train
|
||||
|
||||
```bash
|
||||
|
Reference in New Issue
Block a user