yolov5训练中如何调用use torch.optim.Adam() optimizer

在使用yolov5进行训练时，可以通过修改train.py文件中的代码来调用torch.optim.Adam() optimizer。具体步骤如下：

1. 打开train.py文件，找到以下代码：

# optimizer
pg0, pg1, pg2 = [], [], []  # optimizer parameter groups
for k, v in model.named_parameters():
    if v.requires_grad:
        if '.bias' in k:
            pg2 += [v]  # biases
        elif 'Conv2d.weight' in k:
            pg1 += [v]  # apply weight decay
        else:
            pg0 += [v]  # all else

optimizer = optim.SGD(pg0, lr=hyp['lr0'], momentum=hyp['momentum'], nesterov=True)
optimizer.add_param_group({'params': pg1, 'weight_decay': hyp['weight_decay']})  # add pg1 with weight_decay
optimizer.add_param_group({'params': pg2})  # add pg2 (biases)

2. 将上述代码替换为以下代码：

# optimizer
optimizer = optim.Adam(model.parameters(), lr=hyp['lr0'])

3. 保存修改后的train.py文件，并重新运行训练脚本，即可使用torch.optim.Adam() optimizer进行训练。