feat: 接口对齐 + FP16 量化(第一版优化方案)

- CTRUserDataset → CTRTestSeqDataset,构造参数对齐评测接口
- load_model 签名修正:ckpt_path 作为第一参数
- FP16 量化:model.half() + Embedding 保留 FP32
- move_batch_to_device 自动 FP32→FP16 转换
- 缓存时预转 FP16,减少推理循环开销
- requirements.txt 精简(去除 nvidia-* 包)
- build_env.sh 标准化(set -e + pip install)
- CLAUDE.md 更新开发命令、代码架构、关键接口说明
This commit is contained in:
2026-06-12 20:47:12 +08:00
parent b0ea305ad0
commit 4ee08adff5
4 changed files with 147 additions and 84 deletions
+4 -1
View File
@@ -1,4 +1,7 @@
#!/bin/bash
set -e
# 安装 Python 依赖(评测系统使用阿里云 PyPI 镜像)
pip install -r requirements.txt
echo "build env succeess"
echo "build env success"