PyTorch Lightning设置auto_scale_batch_size='power'无结果输出如何解决

你遗漏了触发自动批大小查找的核心调用步骤，同时可按以下几点逐一排查配置：

• 仅在Trainer初始化时传入auto_scale_batch_size参数不会自动启动搜索流程，你需要在调用trainer.fit()之前，先执行trainer.tune(model)触发搜索逻辑，搜索到的最优批大小会自动更新到你模型的self.batch_size属性上，无需手动赋值。
  参考代码：

trainer = pl.Trainer(default_root_dir=model_dir, 
                     auto_scale_batch_size='power')
# 新增此行触发批大小搜索
trainer.tune(model)
# 搜索完成后再启动正式训练
trainer.fit(model)

• 确认你的模型__init__中如果调用了self.save_hyperparameters()方法，batch_size没有被列入忽略参数列表，否则自动搜索逻辑无法正确更新模型的批大小属性。
• 检查日志输出等级：批大小搜索的相关日志默认是INFO等级，如果你的全局日志等级设置为WARNING及以上，相关输出会被屏蔽。你可以在Trainer初始化时添加enable_progress_bar=True参数，或者调整全局日志等级为INFO，就能看到完整的搜索过程输出。
• 自动批大小搜索是在正式训练启动前运行的，不需要等待训练结束，搜索完成后会直接打印最终得到的最大可用批大小数值，之后才会进入正式训练流程。

内容的提问来源于stack exchange，提问作者Cara Duf