metadata
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
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:
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 |