Skip to content

Commit 85ce29f

Browse files
author
sfluegel
committed
harmonise training configs
1 parent 4591f65 commit 85ce29f

File tree

4 files changed

+5
-7
lines changed

4 files changed

+5
-7
lines changed

configs/training/default_callbacks.yml

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -10,5 +10,3 @@
1010
filename: 'per_{epoch:02d}_{val_loss:.4f}_{val_macro-f1:.4f}_{val_micro-f1:.4f}'
1111
every_n_epochs: 25
1212
save_top_k: -1
13-
#- class_path: chebai.callbacks.epoch_metrics.EpochLevelMacroF1
14-
#class_path: chebai.callbacks.save_config_callback.CustomSaveConfigCallback

configs/training/pretraining_callbacks.yml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,9 +4,9 @@
44
mode: 'min'
55
filename: 'best_{epoch}_{val_loss:.4f}'
66
every_n_epochs: 1
7-
save_top_k: 5
7+
save_top_k: 3
88
- class_path: chebai.callbacks.model_checkpoint.CustomModelCheckpoint
99
init_args:
1010
filename: 'per_{epoch}_{val_loss:.4f}'
11-
every_n_epochs: 5
11+
every_n_epochs: 25
1212
save_top_k: -1

configs/training/pretraining_trainer.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,6 @@ min_epochs: 100
22
max_epochs: 100
33

44
default_root_dir: &default_root_dir logs
5-
logger: wandb_logger.yml
5+
logger: csv_logger.yml
66

77
callbacks: pretraining_callbacks.yml

configs/training/single_class_callbacks.yml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,10 +4,10 @@
44
mode: 'max'
55
filename: 'best_{epoch:02d}_{val_loss:.4f}_{val_f1:.4f}'
66
every_n_epochs: 1
7-
save_top_k: 5
7+
save_top_k: 3
88
- class_path: chebai.callbacks.model_checkpoint.CustomModelCheckpoint
99
init_args:
1010
filename: 'per_{epoch:02d}_{val_loss:.4f}_{val_f1:.4f}'
11-
every_n_epochs: 5
11+
every_n_epochs: 25
1212
save_top_k: -1
1313
# difference to default_callbacks.yml: no macro-f1

0 commit comments

Comments
 (0)