Install the Extension

Launch TorchStudio, and click menu > Settings…

In the IDE Extensions section, install the extension for your IDE.

The following IDEs are supported:

Once the extension installed, you can close TorchStudio and launch your IDE.

Add TorchStudio to your IDE project

Right-click on your project folder then click on New TorchStudio Training
(Spyder: click Open IPython console here and type !torchstudio new)

VS Code, PyCharm, Sublime Text

This will add a TorchStudio Training.tsz file to your project.

Drag your scripts to TorchStudio

To open TorchStudio within your IDE, click on TorchStudio Training.tsz
(Pycharm and Spyder: double-click).

TorchStudio window will appear on top of your IDE (notice the active pin icon pin in TorchStudio’s title bar which maintain the window on top).

Then drag your dataset script from your IDE to TorchStudio (Sublime Text: right-click > Send to TorchStudio). You can alternatively use a dataset provided by TorchStudio.

drag

Click Load in TorchStudio to load it. This allows your to explore the dataset, reformat it and analyze it if needed.

Then create a model tab by clicking on the plus button at the top, and drag your model script from your IDE to TorchStudio (Sublime Text: right-click > Send to TorchStudio). You can alternatively use a model provided by TorchStudio.

Click Build in TorchStudio. This will show you its graph, and allows you to define hyperparameters and a device for the training.

Then click Train to train it.

Export the weights back to your IDE

Once you’ve trained your model, you can export its weights back to your IDE by clicking menu > Export Model… (or press Ctrl+E).

drag

You can then close TorchStudio.

Tutorials