AttributeError: 'Trainer' 对象没有属性 'lr_find'

问题描述 投票:0回答:0
`Trainer object in pytorch lightning cannot use lr_find

trainer = L.Trainer(max_epochs=34, accelerator="auto", devices="auto")

1)
tuner = L.pytorch.tuner.Tuner(trainer)
lr_find_results = tuner.lr_find(model,
                                train_dataloaders=dataloader, 
                                min_lr=0.001, # minimum learning rate
                                max_lr=1.0,   # maximum learning rate
                                early_stop_threshold=None)
2) lr_find_results = trainer.tuner.lr_find(model,
train_dataloaders=dataloader, min_lr=0.001,max_lr=1.0,early_stop_threshold=None) 

出于某种原因,使用方法 2 查找学习率会出现错误 AttributeError: 'Trainer' object has no attribute 'lr_find' ...可能是什么问题?...已经尝试更新 lightning`

python deep-learning pytorch neural-network pytorch-lightning
© www.soinside.com 2019 - 2024. All rights reserved.