| 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 | |