ML Katas

Save and Load TorchScript Model

medium (<30 mins) pytorch torchscript model deployment serialization
this month by E

Convert a trained PyTorch model to TorchScript via tracing and scripting. Save it to disk. Reload and run inference. Compare outputs with the original model.