Skip to content

Commit c9c10ff

Browse files
committed
update code
1 parent 01429e3 commit c9c10ff

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

deepxde/optimizers/paddle/optimizers.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -51,17 +51,17 @@ def get(params, optimizer, learning_rate=None, decay=None, weight_decay=None):
5151
return paddle.optimizer.Adam(
5252
learning_rate=learning_rate, parameters=params, weight_decay=weight_decay
5353
)
54-
elif optimizer == "sgd":
54+
if optimizer == "sgd":
5555
return paddle.optimizer.SGD(
5656
learning_rate=learning_rate, parameters=params, weight_decay=weight_decay
5757
)
58-
elif optimizer == "rmsprop":
58+
if optimizer == "rmsprop":
5959
return paddle.optimizer.RMSProp(
6060
learning_rate=learning_rate,
6161
parameters=params,
6262
weight_decay=weight_decay,
6363
)
64-
elif optimizer == "adamw":
64+
if optimizer == "adamw":
6565
if weight_decay[0] == 0:
6666
raise ValueError("AdamW optimizer requires non-zero weight decay")
6767
return paddle.optimizer.AdamW(

0 commit comments

Comments
 (0)