From 228a38ec09690db6ea0b1f74124f9178c02f2174 Mon Sep 17 00:00:00 2001
From: dcarron <daniel.carron@idiap.ch>
Date: Mon, 1 Jul 2024 11:19:44 +0200
Subject: [PATCH] [segmentation/engine/adabound] Fix typing
---
src/mednet/libs/segmentation/engine/adabound.py | 14 +++++++-------
1 file changed, 7 insertions(+), 7 deletions(-)
diff --git a/src/mednet/libs/segmentation/engine/adabound.py b/src/mednet/libs/segmentation/engine/adabound.py
index bf445ceb..3b37346d 100644
--- a/src/mednet/libs/segmentation/engine/adabound.py
+++ b/src/mednet/libs/segmentation/engine/adabound.py
@@ -51,13 +51,13 @@ class AdaBound(torch.optim.Optimizer):
def __init__(
self,
params: list,
- lr: float | None = 1e-3,
- betas: tuple[float, float] | None = (0.9, 0.999),
- final_lr: float | None = 0.1,
- gamma: float | None = 1e-3,
- eps: float | None = 1e-8,
- weight_decay: float | None = 0,
- amsbound: bool | None = False,
+ lr: float = 1e-3,
+ betas: tuple[float, float] = (0.9, 0.999),
+ final_lr: float = 0.1,
+ gamma: float = 1e-3,
+ eps: float = 1e-8,
+ weight_decay: float = 0,
+ amsbound: bool = False,
):
if not 0.0 <= lr:
raise ValueError(f"Invalid learning rate: {lr}")
--
GitLab