Skip to content

Repository files navigation

Quick Start

preprocessing

python preprocess.py --mode full --cycle_jsons random_generate/output0.json  random_generate/output5.json random_generate/output1.json  random_generate/output6.json random_generate/output2.json  random_generate/output7.json random_generate/output3.json  random_generate/output8.json random_generate/output4.json  random_generate/output9.json
python preprocess.py --cycle_jsons random_generate/output2.json --existing_train_json data/xiangshan/train_data1.json --existing_val_json data/xiangshan/val_data.json --train_json data/xiangshan/tr.json

training

python main.py train --model_type transformer --train_data data/xiangshan/tr1.json --val_data data/xiangshan/val_data.json --experiment_name transformer --epoch 50

incremental training

python main.py incremental --model_path experiments/transformer_v1_20250304_101004/checkpoints/checkpoint_latest.pth

resume

python main.py resume --checkpoint experiments/transformer_v1/checkpoints/checkpoint_epoch_10.pth --additional_epochs 20

About

No description, website, or topics provided.

Resources

Stars

0 stars

Watchers

1 watching

Forks

Releases

Packages

Contributors

Languages