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:
@@ -1,4 +1,7 @@
|
||||
#!/bin/bash
|
||||
set -e
|
||||
|
||||
# 安装 Python 依赖(评测系统使用阿里云 PyPI 镜像)
|
||||
pip install -r requirements.txt
|
||||
|
||||
echo "build env succeess"
|
||||
echo "build env success"
|
||||
|
||||
Reference in New Issue
Block a user