Searched refs:_ScriptLocalOptimizer (Results 1 – 1 of 1) sorted by relevance
40 class _ScriptLocalOptimizer(nn.Module): class99 optim = _ScriptLocalOptimizer(optim_cls, local_params_rref, *args, **kwargs)101 with _ScriptLocalOptimizer.compile_lock: