How to configure PyTorch launcher

For enabling PyTorch launcher you need to add framework: pytorch in launchers section of your configuration file and provide following parameters:

PyTorch launcher config example (demonstrates how to run AlexNet model from torchvision):

launchers:
- framework: pytorch
device: CPU
module: orchvision.models.alexnet
module_kwargs:
pretrained: True
adapter: classification