File size: 1,271 Bytes
08797a4 990a10e 08797a4 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 | ---
language:
- es
library_name: pytorch
pipeline_tag: tabular-regression
tags:
- machine-learning
- regression
- linear-regression
- educational-example
- torchscript
license: mit
---
# PyTorch: exportar en formato TorchScript
Exporta el modelo entrenado en `pytorch-entrenamiento-evaluacion` al formato **TorchScript** (`model_ts.pt`) y lo vuelve a cargar para predecir.
## Cómo correrlo
```bash
python ../pytorch-entrenamiento-evaluacion/train.py # si todavía no está entrenado
python exportar.py
python cargar.py
```
## Usar el modelo desde el Hub
Esta carpeta se puede subir tal cual como repositorio de modelo (desde la raíz: `python publicar_en_hub.py 00-formatos-exportacion/pytorch-exportar-torchscript`). Después, desde cualquier PC:
```bash
python predict_hub.py tu-usuario/casas-pytorch-exportar-torchscript
```
## Qué guarda el formato
| | |
|---|---|
| **¿Qué guarda?** | los pesos y el código del `forward` |
| **¿Cómo se carga?** | `torch.jit.load`, sin la clase. |
| **¿Es seguro abrirlo si viene de un desconocido?** | Sí, no usa pickle. Está *deprecated*: PyTorch recomienda torch.export. |
| **¿Para qué caso?** | proyectos viejos o C++. Para algo nuevo, torch.export |
|