Skip to content

Commit 149d9d7

Browse files
authored
Pass verbosity to predict and DeepLC Retrainer
1 parent 3740720 commit 149d9d7

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

deeplc/deeplc.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -610,7 +610,7 @@ def make_preds_core(
610610
try:
611611
X
612612
ret_preds = mod.predict(
613-
[X, X_sum, X_global, X_hc], batch_size=self.batch_num_tf
613+
[X, X_sum, X_global, X_hc], batch_size=self.batch_num_tf, verbose=int(self.verbose)
614614
).flatten()
615615
except UnboundLocalError:
616616
logger.debug("X is empty, skipping...")
@@ -1170,6 +1170,7 @@ def calibrate_preds(
11701170
freeze_layers=True,
11711171
n_epochs=self.n_epochs,
11721172
freeze_after_concat=1,
1173+
verbose=self.verbose,
11731174
)
11741175

11751176
self.model = models

0 commit comments

Comments
 (0)