Train loop: sgd is Optional[Optimizer], do not pass false

This commit is contained in:
Daniël de Kok 2023-01-20 14:08:42 +01:00
parent 88889bcb02
commit 7753b88e0d

View File

@ -210,7 +210,7 @@ def train_while_improving(
subbatch,
drop=dropout,
losses=losses,
sgd=False, # type: ignore[arg-type]
sgd=None,
exclude=exclude,
annotates=annotating_components,
)