我正在研究图像分类器,并尝试实现循环学习率以获得更好的结果。我正在使用lr_scheduler.LambdaLR来调整训练期间的学习率,但是出现一个错误,我不确定是什么原因造成的。这是我的代码:
lr_find_epochs = 2
start_lr = 1e-7
end_lr = 0.1
# Set up the model, optimizer and loss function for the experiment
optimizer = torch.optim.SGD(model.parameters(), start_lr)
criterion = nn.NLLLoss()
# LR function lambda
lr_lambda = lambda x: math.exp(x * math.log(end_lr / start_lr) / (lr_find_epochs * len( train_loader)))
scheduler = torch.optim.lr_scheduler.LambdaLR(optimizer, lr_lambda=lr_lambda)
lr_scheduler.py中的错误。
问题是由此行引起的
scheduler = torch.optim.lr_scheduler.LambdaLR(optimizer, lr_lambda=lr_lambda)
由于错误提示,您正在尝试在赋值之前参考该值,即lambda函数以自身作为参数调用,当前未分配给任何东西。结果,在lr_scheduler.py
也许您想将其他内容传递给lambda函数。