Skip to content

vessl-ai/examples

Repository files navigation

VESSL examples

This repository contains VESSL examples. If you want to learn more about VESSL please follow the quick start documentation.

Use hyperparameters on VESSL

Hyperparameters are automatically add to the container as environment variables with the given key and value. If you want to use them at runtime, then append them to the start command as follows.

# Add learning_rate as a hyperparameter
python main.py --learning-rate $learning_rate

You can now take the desired type of argument in Python script at runtime.

import argparse

parser = argparse.ArgumentParser()
parser.add_argument('--learning-rate', type=float, default=0.01)
args = parser.parse_args()

Try out VESSL examples