Skip to content

Commit c35b277

Browse files
committed
update AC-discovery comments
1 parent 8484141 commit c35b277

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

examples/AC-discovery.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -58,8 +58,10 @@ def f_model(u_model, var, x, t):
5858
model = DiscoveryModel()
5959
model.compile(layer_sizes, f_model, X, u_star, params,
6060
col_weights=col_weights) # baseline discovery approach can be done by simply removing the col_weights arg
61+
62+
# an example as to how one could modify an optimizer, in this case the col_weights optimizer
6163
model.tf_optimizer_weights = tf.keras.optimizers.Adam(lr=0.005,
62-
beta_1=.95) # an example as to how one could modify an optimizer, in this case the col_weights optimizer
64+
beta_1=.95)
6365

6466
# train loop
6567
model.fit(tf_iter=10000)

0 commit comments

Comments
 (0)