Skip to content

Commit 0f2e903

Browse files
committed
settling on final default checkpoint param values
1 parent fa6641a commit 0f2e903

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

examples/ADRP/adrp_bmkdata_model.txt

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,14 @@ test_data = 'ml.ADRP_6W02_A_1_H.Orderable_zinc_db_enaHLL.sorted.4col.dd.dedup.t
88
train_data = 'ml.ADRP_6W02_A_1_H.Orderable_zinc_db_enaHLL.sorted.4col.dd.dedup.train.parquet'
99
val_data = 'ml.ADRP_6W02_A_1_H.Orderable_zinc_db_enaHLL.sorted.4col.dd.dedup.val.parquet'
1010

11+
# for checkpointing
12+
ckpt_restart_mode = 'off'
13+
ckpt_save_best = True
14+
ckpt_save_best_metric = 'val_loss'
15+
ckpt_skip_epochs = 50
16+
ckpt_save_interval = 500
17+
ckpt_keep_limit = 1
18+
1119
base_name = 'ADRP_6W02_A_1_H'
1220
model_name = 'adrp'
1321
dense = [250, 125, 60, 30]

0 commit comments

Comments
 (0)