Add NeMo backend probe and architecture validator

This commit is contained in:
2026-07-01 20:46:45 +08:00
parent 11308f95c4
commit cdd8e2f85b
3 changed files with 122 additions and 0 deletions
+11
View File
@@ -0,0 +1,11 @@
# NVIDIA NeMo Backend
本项目的训练后端目标是 NVIDIA 官方 NeMo/Megatron 栈,而不是旧的手写 PyTorch trainer。
默认镜像:
```text
nvcr.io/nvidia/nemo:26.06
```
如果 g0033 拉取 NGC 镜像受限,可以先在可联网机器拉取后导入,或在本机配置 Docker daemon 代理。训练脚本只依赖 `/mnt/beegfs` 挂载,不把模型权重、数据 shard 或日志提交到 git。