Skip to content

Commit f1f288c

Browse files
Chao1Hanzhuyuhua-v
andauthored
Add default param for create_optimizer_lars (#4648) (#4667)
* Add default param for create_optimizer_lars * flake8 --------- Co-authored-by: zhuyuhua-v <[email protected]>
1 parent df87693 commit f1f288c

File tree

1 file changed

+7
-1
lines changed
  • intel_extension_for_pytorch/optim

1 file changed

+7
-1
lines changed

intel_extension_for_pytorch/optim/_lars.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,13 @@
1212

1313

1414
def create_optimizer_lars(
15-
model, lr, momentum, weight_decay, bn_bias_separately, epsilon, fused
15+
model,
16+
lr,
17+
momentum,
18+
weight_decay,
19+
bn_bias_separately,
20+
epsilon,
21+
fused=True,
1622
):
1723
if bn_bias_separately:
1824
optimizer = Lars(

0 commit comments

Comments
 (0)