在Jupyter Notebook上运行Aim UI

在笔记本中运行以下代码来安装Aim：

!pip install aim

接下来，初始化一个新的运行并保存一些超参数：

from aim import Run

run = Run()

run['hparams'] = {
    'learning_rate': 0.001,
    'batch_size': 32,
}

注意

训练结束后不要忘记调用 run.finalize()。

在使用Run跟踪运行后，在笔记本中运行以下命令以启动Aim UI界面：

1. 加载Aim扩展以支持笔记本环境:

%load_ext aim

1. 运行 %aim up 在笔记本中打开 Aim UI：

%aim up