计算模型大小 model = XXModel() total_params = sum(p.numel() for p in model.parameters()) total_params_mb = total_params / 1e6 print(f"Total parameters: {total_params_mb:.2f} MB")