From 1ee28ef835755510497425e94564ac04c4a9d851 Mon Sep 17 00:00:00 2001 From: Burak Date: Mon, 2 Oct 2023 16:07:05 +0200 Subject: [PATCH] SemanticSegmentationTask Signature in README Update SemanticSegmentationTask's signature in README.md after 0.5 --- README.md | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index 7ab9840d040..c33f4a55034 100644 --- a/README.md +++ b/README.md @@ -136,8 +136,8 @@ task = SemanticSegmentationTask( num_classes=2, loss="ce", ignore_index=None, - learning_rate=0.1, - learning_rate_schedule_patience=6, + lr=0.1, + patience=6, ) trainer = Trainer(default_root_dir="...")