Searched defs:optim_kwargs (Results 1 – 3 of 3) sorted by relevance
209 self, model: torch.nn.Module, **optim_kwargs
64 def __init__(self, optim_cls: Type, params, *optim_args, **optim_kwargs) -> None:
4792 optim_kwargs, argument