【WanB可视化自动调参学习】第三章：初始化sweep controller和启动sweep agent

初始化sweep controller

sweep_id = wandb.sweep(sweep_config, project = config.project_name)

启动sweep agent

将模型训练相关的全部代码整理为一个train函数。

准备

一个完整的train包含以下几个部分：

1. 数据加载

dl_train,dl val = create_dataloaders(config)

2. 模型建立

model = create_net(config);

3. 训练循环

model = train_epoch(model, dl_train, optimizer)

4. 评估循环

val_acc = eval_epoch(model, dl_val)

开始

# 该agent随机搜索，尝试5次
wandb.agent(sweep_id, train, count = 5)