mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-10-24 09:17:58 +00:00
648f806315
* add example of self-supervised SimCLR training * simclr v2, replace nvidia dali dataloader * updated * sync to latest code writing style * sync to latest code writing style and modify README * detail README & standardize dataset path
7 lines
226 B
Bash
7 lines
226 B
Bash
#!/usr/bin/env sh
|
|
|
|
## phase 1: self-supervised training
|
|
python -m torch.distributed.launch --nproc_per_node 1 train_simclr.py
|
|
|
|
## phase 2: linear evaluation
|
|
python -m torch.distributed.launch --nproc_per_node 1 train_linear.py |