File size: 564 Bytes
6534252 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 |
# 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 |