File size: 686 Bytes
b7720f0 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 | install:
python -m pip install -r requirements.txt
train:
python train.py --qlora --bf16 --num_train_epochs 3 --train_batch_size 1 --gradient_accumulation_steps 16
infer:
python infer.py --model_dir outputs/GravityLLM-Qwen2.5-1.5B-S9 --input_json examples/sample_input.json --validate --output_json outputs/sample_prediction.json
eval:
python evaluate.py --model_dir outputs/GravityLLM-Qwen2.5-1.5B-S9 --data_file data/valid.jsonl --report_path reports/eval_report.json
validate-example:
python tools/validate_scene.py schemas/scene.schema.json examples/sample_output.json
synthetic-data:
python tools/make_synthetic_dataset.py --output data/synthetic_train.jsonl --count 50
|