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