确定模型层数

python -c "import torch; ckpt = torch.load('full_sft_512.pth', map_location='cpu'); print('模型键:', ckpt.keys() if isinstance(ckpt, dict) else 'direct tensor'); print('\n层数统计:'); [print(k) for k in (ckpt.keys() if isinstance(ckpt, dict) else [])]"

50 系显卡 cuda 和 pytorch 不兼容问题，升级到最新 cu129

pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129