Commit ddfcdc6f authored by Artem Pulkin's avatar Artem Pulkin

ml_util: switch to LBFGS default

parent 62d95771
Pipeline #43692 passed with stage
in 2 minutes and 23 seconds
......@@ -456,9 +456,9 @@ class SimpleClosure:
if criterion is None:
criterion = torch.nn.MSELoss()
if optimizer is None:
optimizer = torch.optim.Adam
optimizer = torch.optim.LBFGS
if optimizer_kwargs is None:
optimizer_kwargs = dict(lr=1e-2)
optimizer_kwargs = dict(line_search_fn="strong_wolfe")
if optimizer_kwargs is None:
optimizer_kwargs = {}
if loss_function_kwargs is None:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment