Skip to content

Commit 66e2e87

Browse files
committed
[skip actions] pre-commit fix
1 parent 1e89217 commit 66e2e87

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

src/pytorch_tabular/tabular_model.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -557,7 +557,9 @@ def train(
557557
if self.config.auto_lr_find and (not self.config.fast_dev_run):
558558
logger.info("Auto LR Find Started")
559559
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.")
560+
logger.info(
561+
f"Suggested LR: {result['lr_find'].suggestion()}. For plot and detailed analysis, use `find_learning_rate` method."
562+
)
561563
# Parameters in models needs to be initialized again after LR find
562564
self.model.data_aware_initialization(self.datamodule)
563565
self.model.train()

0 commit comments

Comments
 (0)