Run and version a notebook as a standard Valohai execution.

Every time you stop a notebook execution Valohai will save the edited notebooks as outputs, so you can easily continue from where you left how.

However, it's also a good practice to use the "Run Remote" feature to run a separate execution that executes the notebook from first to last cell and versions it as a standard Valohai execution. This will make it easier for other team members to read and leverage your notebook, as they can rely that the execution installs the right dependencies and the notebook works by just running the cells in order.

Start by adding a new cell that defines the Valohai step and it's properties.

Check out our Getting started with Valohai guide for instructions on how to define steps with valohai-utils.

Below you'll find an example:

import valohai

valohai.prepare(
    step='train-model',
    image='tensorflow/tensorflow:2.6.0',
    default_inputs={
        'input': 'https://valohaidemo.blob.core.windows.net/mnist/mnist.npz'
    },
    default_parameters={
        'learning_rate': 0.001,
        'epoch': 10
    }
)