We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 1e89217 commit 66e2e87Copy full SHA for 66e2e87
src/pytorch_tabular/tabular_model.py
@@ -557,7 +557,9 @@ def train(
557
if self.config.auto_lr_find and (not self.config.fast_dev_run):
558
logger.info("Auto LR Find Started")
559
result = self.trainer.tune(self.model, train_loader, val_loader)
560
- logger.info(f"Suggested LR: {result['lr_find'].suggestion()}. For plot and detailed analysis, use `find_learning_rate` method.")
+ logger.info(
561
+ f"Suggested LR: {result['lr_find'].suggestion()}. For plot and detailed analysis, use `find_learning_rate` method."
562
+ )
563
# Parameters in models needs to be initialized again after LR find
564
self.model.data_aware_initialization(self.datamodule)
565
self.model.train()
0 commit comments