When saving or loading a trained model, the model should be in eval() mode rather than train() mode. An alternative approach for saving a PyTorch model is to use ONNX (Open Neural Network Exchange).

The Data Science Lab. Multi-Class Classification Using PyTorch: Defining a Network. Dr. James McCaffrey of Microsoft Research explains how to define a network in installment No. 2 of his four-part ...

The logical place to train a new model is on a cloud-hosted platform, such as Azure’s Machine Learning studio. This can get expensive, requiring large virtual machines to host your models and a ...