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, subbatch,
drop=dropout, drop=dropout,
losses=losses, losses=losses,
sgd=False, # type: ignore[arg-type] sgd=None,
exclude=exclude, exclude=exclude,
annotates=annotating_components, annotates=annotating_components,
) )