26 lines
608 B
Batchfile
26 lines
608 B
Batchfile
@echo off
|
|
REM 快速测试训练 - 只加载20个episodes
|
|
echo ============================================================
|
|
echo 快速测试模式
|
|
echo ============================================================
|
|
echo.
|
|
echo 配置说明:
|
|
echo - 只加载前20个episodes (快速测试)
|
|
echo - Batch Size: 512
|
|
echo - 混合精度训练: 启用
|
|
echo.
|
|
echo 开始测试训练...
|
|
echo ============================================================
|
|
echo.
|
|
|
|
python train.py ^
|
|
--max_episodes 20 ^
|
|
--batch_size 512 ^
|
|
--update_frequency 5 ^
|
|
--num_epochs 5 ^
|
|
--use_amp ^
|
|
--device cuda
|
|
|
|
pause
|
|
|