- 博客(1)
- 收藏
- 关注
原创 pytorch_lighting中Trainer的auto_lr_find不起作用
若有原callback设置需要同时导入,只需要将其放入Trainer的callbacks的序列中。之后再运行trainer.fit(),就会开始自动寻找最佳学习率。需要首先导入LearningRateFinder。后新建一类再传给Trainer的callback。设置Trainer的callbacks中的。
2024-04-02 13:34:50 210
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人