@echo off
echo ========================================
echo NOVEL AI MODEL - BENCHMARK TRAINING
echo ========================================
echo.
echo This will train the model on GPU and run benchmark evaluations.
echo Training will take approximately 1-2 hours for 5 epochs.
echo.
echo Configuration:
echo   - Epochs: 5
echo   - Batch Size: 4
echo   - Learning Rate: 1e-3
echo   - Device: GPU
echo   - Evaluation: Every 2 epochs
echo.
echo Press any key to start training...
pause > nul

python training/train_with_eval_fixed2.py --train_data ./data/sample_train.txt --val_data ./data/sample_valid.txt --test_data ./data/sample_test.txt --test_names sample_test --epochs 5 --batch_size 4 --learning_rate 1e-3 --eval_interval 2 --output_dir ./checkpoints/benchmark_run --save_best

echo.
echo ========================================
echo Training Complete!
echo ========================================
echo.
echo Results saved in: ./checkpoints/benchmark_run/
echo.
pause