275 lines
14 KiB
HTML
275 lines
14 KiB
HTML
<!DOCTYPE html>
|
||
<html lang="zh-CN">
|
||
<head>
|
||
<meta charset="UTF-8">
|
||
<meta name="viewport" content="width=device-width, initial-scale=1.0">
|
||
<title>MiniMind Training Lab</title>
|
||
<link rel="stylesheet" href="/static/css/style.css">
|
||
</head>
|
||
<body>
|
||
<div class="header">
|
||
<img src="/static/images/logo2.png" alt="MiniMind Logo" class="logo">
|
||
<h1>MiniMind Training Lab</h1>
|
||
</div>
|
||
|
||
<div class="tabs">
|
||
<button class="tab active" onclick="openTab(event, 'train')">开始训练</button>
|
||
<button class="tab" onclick="openTab(event, 'processes')">训练进程</button>
|
||
<button class="tab" onclick="openTab(event, 'logfiles')">日志文件</button>
|
||
</div>
|
||
|
||
<div id="train" class="tab-content">
|
||
<div class="form-container">
|
||
<h2 class="section-title">选择训练类型并配置参数</h2>
|
||
<form id="train-form" method="post" action="/train">
|
||
<!-- 基础训练参数 -->
|
||
<div class="parameter-card">
|
||
<h3 class="card-title">基础训练参数</h3>
|
||
<div class="parameter-content">
|
||
<div class="form-group">
|
||
<label for="train_type">训练类型：</label>
|
||
<select id="train_type" name="train_type" required>
|
||
<option value="pretrain">Pretrain</option>
|
||
<option value="sft">SFT - Full</option>
|
||
<option value="lora">SFT - Lora</option>
|
||
<option value="dpo">RL - DPO</option>
|
||
<option value="ppo">RL - PPO</option>
|
||
</select>
|
||
</div>
|
||
<div class="form-group">
|
||
<label for="epochs">训练轮数：</label>
|
||
<input type="number" id="epochs" name="epochs" min="1" required>
|
||
</div>
|
||
<div class="form-group">
|
||
<label for="batch_size">Batch Size：</label>
|
||
<input type="number" id="batch_size" name="batch_size" min="1" required>
|
||
</div>
|
||
<div class="form-group">
|
||
<label for="learning_rate">学习率：</label>
|
||
<input type="text" id="learning_rate" name="learning_rate" required>
|
||
</div>
|
||
<div class="form-group">
|
||
<label for="log_interval">日志打印间隔：</label>
|
||
<input type="number" id="log_interval" name="log_interval" min="1" required>
|
||
</div>
|
||
<div class="form-group">
|
||
<label for="data_path">数据路径：</label>
|
||
<input type="text" id="data_path" name="data_path" required>
|
||
</div>
|
||
</div>
|
||
</div>
|
||
|
||
<!-- 强化学习参数 -->
|
||
<div class="parameter-card dpo" style="display: none;">
|
||
<h3 class="card-title">强化学习参数 (DPO)</h3>
|
||
<div class="parameter-content">
|
||
<div class="form-group">
|
||
<label for="beta">DPO Beta 参数：</label>
|
||
<input type="text" id="beta" name="beta" placeholder="0.1">
|
||
</div>
|
||
</div>
|
||
</div>
|
||
|
||
<!-- PPO强化学习参数 -->
|
||
<div class="parameter-card ppo" style="display: none;">
|
||
<h3 class="card-title">强化学习参数 (PPO)</h3>
|
||
<div class="parameter-content">
|
||
<div class="form-group">
|
||
<label for="clip_epsilon">PPO剪切系数：</label>
|
||
<input type="text" id="clip_epsilon" name="clip_epsilon" placeholder="0.2">
|
||
</div>
|
||
<div class="form-group">
|
||
<label for="vf_coef">价值函数系数：</label>
|
||
<input type="text" id="vf_coef" name="vf_coef" placeholder="0.1">
|
||
</div>
|
||
<div class="form-group">
|
||
<label for="kl_coef">KL散度惩罚系数：</label>
|
||
<input type="text" id="kl_coef" name="kl_coef" placeholder="0.01">
|
||
</div>
|
||
<div class="form-group">
|
||
<label for="reasoning">是否使用Reasoning模式：</label>
|
||
<select id="reasoning" name="reasoning">
|
||
<option value="0">否</option>
|
||
<option value="1">是</option>
|
||
</select>
|
||
</div>
|
||
<div class="form-group">
|
||
<label for="update_old_actor_freq">更新旧Actor频率：</label>
|
||
<input type="number" id="update_old_actor_freq" name="update_old_actor_freq" placeholder="10" min="1">
|
||
</div>
|
||
<div class="form-group">
|
||
<label for="reward_model_path">奖励模型路径：</label>
|
||
<input type="text" id="reward_model_path" name="reward_model_path" placeholder="path/to/reward/model">
|
||
</div>
|
||
</div>
|
||
</div>
|
||
|
||
<!-- 模型结构参数 -->
|
||
<div class="parameter-card">
|
||
<h3 class="card-title">模型结构参数</h3>
|
||
<div class="parameter-content">
|
||
<div class="form-group">
|
||
<label for="hidden_size">隐藏层维度：</label>
|
||
<input type="number" id="hidden_size" name="hidden_size" min="128" required>
|
||
</div>
|
||
<div class="form-group">
|
||
<label for="num_hidden_layers">隐藏层数量：</label>
|
||
<input type="number" id="num_hidden_layers" name="num_hidden_layers" min="1" required>
|
||
</div>
|
||
<div class="form-group">
|
||
<label for="max_seq_len">最大序列长度：</label>
|
||
<input type="number" id="max_seq_len" name="max_seq_len" min="64" required>
|
||
</div>
|
||
<div class="form-group">
|
||
<label for="use_moe">是否使用MoE架构：</label>
|
||
<select id="use_moe" name="use_moe">
|
||
<option value="0">否</option>
|
||
<option value="1">是</option>
|
||
</select>
|
||
</div>
|
||
</div>
|
||
</div>
|
||
|
||
<!-- 模型保存与恢复 -->
|
||
<div class="parameter-card">
|
||
<h3 class="card-title">模型保存与恢复</h3>
|
||
<div class="parameter-content">
|
||
<div class="form-group">
|
||
<label for="save_dir">模型保存目录：</label>
|
||
<input type="text" id="save_dir" name="save_dir" required>
|
||
</div>
|
||
<div class="form-group">
|
||
<label for="save_interval">模型保存间隔：</label>
|
||
<input type="number" id="save_interval" name="save_interval" min="1" required>
|
||
</div>
|
||
<div class="form-group pretrain-sft">
|
||
<label for="save_weight">保存权重前缀名：</label>
|
||
<input type="text" id="save_weight" name="save_weight">
|
||
</div>
|
||
<div class="form-group lora">
|
||
<label for="lora_name">LoRA权重名称：</label>
|
||
<input type="text" id="lora_name" name="lora_name">
|
||
</div>
|
||
<div class="form-group from-weight">
|
||
<label for="from_weight">基于哪个权重训练：</label>
|
||
<input type="text" id="from_weight" name="from_weight">
|
||
</div>
|
||
<div class="form-group">
|
||
<div class="checkbox-group">
|
||
<input type="checkbox" id="from_resume" name="from_resume" value="1">
|
||
<label for="from_resume">是否自动检测&续训</label>
|
||
</div>
|
||
</div>
|
||
</div>
|
||
</div>
|
||
|
||
<!-- 其他设置 -->
|
||
<div class="parameter-card">
|
||
<h3 class="card-title">其他设置</h3>
|
||
<div class="parameter-content">
|
||
<div class="form-group">
|
||
<label for="training_mode">训练方式：</label>
|
||
<select id="training_mode" name="training_mode" required>
|
||
<option value="single_gpu">单卡训练</option>
|
||
<option value="multi_gpu">多卡训练</option>
|
||
<option value="cpu">CPU训练</option>
|
||
</select>
|
||
</div>
|
||
<div class="form-group" id="single-gpu-selection">
|
||
<label for="device">GPU序号：</label>
|
||
<input type="number" id="device" name="device" min="0" max="{{ gpu_count|default(1) - 1 }}" value="0" required>
|
||
</div>
|
||
<div class="form-group" id="multi-gpu-selection" style="display: none;">
|
||
<label for="gpu_num">多卡并行数：</label>
|
||
<input type="number" id="gpu_num" name="gpu_num" min="1" max="{{ gpu_count|default(1) }}" value="{{ gpu_count|default(1) }}" required>
|
||
<span class="hint-text">(可用GPU数量: {{ gpu_count|default(0) }})</span>
|
||
</div>
|
||
<div class="form-group">
|
||
<label for="train_monitor">训练监控：</label>
|
||
<select id="train_monitor" name="train_monitor">
|
||
<option value="none">无监控</option>
|
||
<option value="wandb">使用WandB/SwanLab</option>
|
||
</select>
|
||
</div>
|
||
</div>
|
||
</div>
|
||
|
||
<script>
|
||
// 检查是否支持GPU并显示对应的选择器
|
||
// 使用Flask模板变量传递GPU信息
|
||
var hasGpu = {{ has_gpu|default(false)|tojson|safe }};
|
||
var gpuCount = {{ gpu_count|default(0)|tojson|safe }};
|
||
|
||
// 根据是否有GPU设置默认训练方式
|
||
var trainingModeSelect = document.getElementById('training_mode');
|
||
var singleGpuSelection = document.getElementById('single-gpu-selection');
|
||
var multiGpuSelection = document.getElementById('multi-gpu-selection');
|
||
|
||
// 设置默认值并初始化UI
|
||
function initUI() {
|
||
if (!trainingModeSelect) return;
|
||
|
||
if (!hasGpu) {
|
||
// 没有GPU时，默认选择CPU训练
|
||
trainingModeSelect.value = 'cpu';
|
||
if (singleGpuSelection) singleGpuSelection.style.display = 'none';
|
||
if (multiGpuSelection) multiGpuSelection.style.display = 'none';
|
||
} else {
|
||
// 设置GPU数量的默认值
|
||
var gpuNumInput = document.getElementById('gpu_num');
|
||
if (gpuNumInput && gpuCount > 0) {
|
||
gpuNumInput.value = gpuCount;
|
||
}
|
||
}
|
||
|
||
// 初始化显示状态
|
||
updateSelectionVisibility();
|
||
}
|
||
|
||
// 更新选择器可见性
|
||
function updateSelectionVisibility() {
|
||
if (!trainingModeSelect) return;
|
||
|
||
var selectedMode = trainingModeSelect.value;
|
||
|
||
if (singleGpuSelection) {
|
||
singleGpuSelection.style.display = selectedMode === 'single_gpu' ? 'block' : 'none';
|
||
}
|
||
if (multiGpuSelection) {
|
||
multiGpuSelection.style.display = selectedMode === 'multi_gpu' ? 'block' : 'none';
|
||
}
|
||
}
|
||
|
||
// 添加训练方式切换事件监听器
|
||
if (trainingModeSelect) {
|
||
trainingModeSelect.addEventListener('change', updateSelectionVisibility);
|
||
}
|
||
|
||
// 初始化UI
|
||
window.addEventListener('DOMContentLoaded', initUI);
|
||
</script>
|
||
|
||
<div class="submit-container">
|
||
<button type="submit">开始训练</button>
|
||
</div>
|
||
</form>
|
||
</div>
|
||
</div>
|
||
|
||
<div id="processes" class="tab-content hidden">
|
||
<h2 class="section-title">训练进程列表</h2>
|
||
<div id="process-list">
|
||
<!-- 进程列表将通过JavaScript动态加载 -->
|
||
</div>
|
||
</div>
|
||
|
||
<div id="logfiles" class="tab-content hidden">
|
||
<h2 class="section-title">日志文件列表</h2>
|
||
<div id="logfiles-list">
|
||
<!-- 日志文件列表将通过JavaScript动态加载 -->
|
||
</div>
|
||
</div>
|
||
|
||
<script src="/static/js/script.js"></script>
|
||
</body>
|
||
</html> |