10 lines
221 B
(Stored with Git LFS)
Plaintext
10 lines
221 B
(Stored with Git LFS)
Plaintext
# generate with torch
|
|
python - <<'EOF'
|
|
import torch, torch.nn as nn
|
|
import torch.onnx
|
|
|
|
model = nn.Linear(3, 1)
|
|
x = torch.randn(1, 3)
|
|
torch.onnx.export(model, x, "model.onnx", input_names=["in"], output_names=["out"])
|
|
EOF
|