Skip to content

Commit a432a8a

Browse files
committed
check if ext params is not empty
1 parent 64c59a3 commit a432a8a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

deepxde/model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -403,7 +403,7 @@ def _compile_jax(self, lr, loss_fn, decay):
403403
self.net.params = self.net.init(key, self.data.test()[0])
404404
external_trainable_variables_val = [
405405
var.value for var in self.external_trainable_variables
406-
]
406+
] if self.external_trainable_variables else []
407407
self.params = [self.net.params, external_trainable_variables_val]
408408
# TODO: learning rate decay
409409
self.opt = optimizers.get(self.opt_name, learning_rate=lr)

0 commit comments

Comments
 (0)