| # Test script to verify dataset loading and model setup | |
| python examples/glen_phase2/train_glen.py ` | |
| --output_dir logs/model_glen_vault/test_setup ` | |
| --model_name_or_path logs/model_glen_vault/GLEN_P1_base ` | |
| --per_device_train_batch_size 2 ` | |
| --per_device_eval_batch_size 1 ` | |
| --gradient_accumulation_steps 4 ` | |
| --test100 1 ` | |
| --num_train_epochs 1 ` | |
| --logging_steps 10 ` | |
| --overwrite_output_dir ` | |
| --do_eval False ` | |
| --gpu_memory_threshold 0.85 ` | |
| --gpu_check_interval 10 ` | |
| --fp16 True ` | |
| --gradient_checkpointing True |