search

isaacmg / model_agnostic_prediction

Peelout model agnostic is a class for easily performing production pipeline predictions.
8 stars 2 forks source link

Unit tests for PyTorch #8

Open isaacmg opened 6 years ago

isaacmg commented 6 years ago

Need to add unit tests to test PyTorch model loading and exporting of model to ONNX.