Pytorch Examples

PyTorch - MNIST Training

pytorch_mnist.py is an example of PyTorch MNIST training integration.

Pre-populated examples ready to enqueue

A self-hosted Trains Server installs with the example experiments located in the trains repository, examples folder

The experiment results tabs that contain the features in this example are the following:

  • HYPER PARAMETERS - Command line arguments.
  • ARTIFACTS
    • Input model, if the experiment executes for a second time, then this is a link to the input model details on the Projects page, models table.
    • Creating experiment (experiment creating this model). This is a link to that experiment's details on the Projects page, experiments table.
    • Output model is a link to the output model details in the Projects page, models table.
  • RESULTS
  • LOG - Console standard output/error

PyTorch and Matplotlib - Testing Style Transfer

pytorch_matplotlib.py is an example of connecting the neural style transfer from the official PyTorch tutorial to TRAINS. Neural-Style, or Neural-Transfer, allows you to take an image and reproduce it with a new artistic style. The algorithm takes three images, an input-image, a content-image, and a style-image, and changes the input-image to resemble the content of the content-image and the artistic style of the style-image.

The experiment results tabs that contain the features in this example are the following:

  • HYPER PARAMETERS - Command line arguments.
  • ARTIFACTS
    • Input model, if the experiment executes for a second time, then this is a link to the input model details on the Projects page, models table.
    • Output model is a link to the output model details in the Projects page, models table.
  • RESULTS
    • DEBUG SAMPLES - Input image, input style images, an output transferred style image.
    • LOG - Console standard output/error.

PyTorch with TensorBoard - MNIST Train

pytorch_tensorboard.py is an example of PyTorch MNIST training running with TensorBoard.

The experiment results tabs that contain the features in this example are the following:

  • HYPER PARAMETERS - Command line arguments.
  • ARTIFACTS
    • Input model, if the experiment executes for the second time, then this is a link to the input model details on the Projects page, models table.
    • Creating experiment (experiment creating this model). This is a link to that experiment's details on the Projects page, experiments table.
    • Output model (a link to the output model details on the Projects page, models table).
  • RESULTS
    • SCALARS - Train and test loss scalars.
    • LOG - Console standard output/error.

PyTorch with TensorBoardX - MNIST Train

pytorch_tensorboardX.py is an example of PyTorch MNIST training running with TensorBoardX.

The experiment results tabs that contain the features in this example are the following:

  • HYPER PARAMETERS - Command line arguments.
  • ARTIFACTS
    • Input model, if the experiment executes for a second time, then this is a link to the input model details on the Projects page, models table.
    • Creating experiment (experiment creating this model). This is a link to that experiment's details on the Projects page, experiments table.
    • Output model (a link to the output model details on the Projects page, models table).
  • RESULTS
    • SCALARS - Train and test loss scalars.
    • LOG - Console standard output/error.

PyTorch Lightning

See [PyTorch Lightning] in the "Integrations" sections.