README.md 349 Bytes
Newer Older
Zihan Wang's avatar
Zihan Wang committed
1
2
3
4
Code for Roformer.

Run with
```bash
5
6
DATA_PATH=???
OUTPUT_DIR=???
Zihan Wang's avatar
Zihan Wang committed
7
8
9
python3 train.py \
  --experiment=roformer/pretraining \
  --config_file=experiments/roformer_base.yaml \
10
  --params_override="task.validation_data.input_path=${DATA_PATH},runtime.distribution_strategy=tpu" \
Zihan Wang's avatar
Zihan Wang committed
11
  --tpu=local \
12
  --model_dir=${OUTPUT_DIR} \
Zihan Wang's avatar
Zihan Wang committed
13
  --mode=train_and_eval
14
```