aim_run_tracking_metrics_and_hyperparameters_quickstart.py

1import aim
2import random
3
4# Initialize a new Run
5run = aim.Run()
6
7# Log hyperparameters
8run["hparams"] = {
9    "learning_rate": 0.001,
10    "batch_size": 32,
11}
12
13# Log metrics to visualize performance
14for i in range(100):
15    run.track(i, name='loss', step=i, context={ "subset": "train" })
16    run.track(i, name='acc', step=i, context={ "subset": "train" })
17    
18    # Tracking a validation metric
19    if i % 10 == 0:
20        run.track(i, name='loss', step=i, context={ "subset": "val" })