Skip to content

Commit

Permalink
fix
Browse files Browse the repository at this point in the history
  • Loading branch information
nmcardoso committed Aug 19, 2023
1 parent 5e56956 commit 149f268
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion mergernet/estimators/base.py
Original file line number Diff line number Diff line change
Expand Up @@ -101,7 +101,7 @@ def compile_model(
metrics: list = [],
label_smoothing: float = 0.0
):
hp_metrics = [self.get_metric(m) for m in self.hp.get('metrics', []) if m is not None]
hp_metrics = [self.get_metric(m) for m in self.hp.get('metrics', default=[]) if m is not None]
tf_model.compile(
optimizer=optimizer,
loss=tf.keras.losses.CategoricalCrossentropy(from_logits=True, label_smoothing=label_smoothing),
Expand Down

0 comments on commit 149f268

Please sign in to comment.