Library Search
结果页PyTorch 训练 → ONNX 导出 → 推理加速
在 技术栈 中找到 3 个匹配结果。
技术栈
技术栈
3 个 PyTorch # ============================================================
# 1. 数据加载与预处理
# ============================================================
transform = transforms.Compose([
transforms.ToTensor(),… 4% ONNX # ONNX 完整流程：PyTorch 导出 → ONNX Runtime 推理
## 目标
演示标准工作流：PyTorch 训一个 ResNet → 导出为 ONNX → ONNX Runtime 推理 → 验证一致性 + 性能对比。
## 完整代码
```python
import torch
import torchvision.models as models
import num… 3% HuggingFace Transformers 1. 为什么 `attention_mask` 对 padding 位置设 0？pytorch 中 mask 值为 0 还是 -inf？
2. LoRA 为什么只训练 Q/V 投影而不训练 FFN？r=8 的 rank 如何影响效果？
3. BERT 和 GPT 的 tokenizer 有何本质区别？为什么 GPT 不能直接用 BERT tokenizer？ 2%