37 lines
1.1 KiB
Batchfile
37 lines
1.1 KiB
Batchfile
@echo off
|
||
REM GPU优化训练 - 推荐配置
|
||
REM 针对RTX 2060和你的数据规模优化
|
||
|
||
echo ============================================================
|
||
echo GPU优化训练 - 快速配置
|
||
echo ============================================================
|
||
echo.
|
||
echo 配置说明:
|
||
echo - Batch Size: 4096 (超大batch，GPU满载极限)
|
||
echo - Hidden Dim: 768 (适中网络，保证模型质量)
|
||
echo - Update Frequency: 100 (减少更新次数提速)
|
||
echo - Epochs Per Update: 2 (快速迭代，用更多epoch补偿)
|
||
echo - Num Epochs: 30 (增加总轮数，保证训练充分)
|
||
echo - Buffer Capacity: 150000 (超大缓冲区充分利用显存)
|
||
echo - 混合精度训练: 启用(AMP加速)
|
||
echo - 设备: CUDA
|
||
echo - 优化目标: 极限速度 + 模型质量保证
|
||
echo.
|
||
echo 开始训练...
|
||
echo ============================================================
|
||
echo.
|
||
|
||
python train.py ^
|
||
--batch_size 2048 ^
|
||
--hidden_dim 768 ^
|
||
--epochs_per_update 2 ^
|
||
--update_frequency 100 ^
|
||
--buffer_capacity 150000 ^
|
||
--use_amp ^
|
||
--device cuda ^
|
||
--num_epochs 10 ^
|
||
--max_episodes 1000
|
||
|
||
pause
|
||
|