diff options
Diffstat (limited to 'training/conf/config.yaml')
| -rw-r--r-- | training/conf/config.yaml | 10 |
1 files changed, 8 insertions, 2 deletions
diff --git a/training/conf/config.yaml b/training/conf/config.yaml index 5897d87..9ed366f 100644 --- a/training/conf/config.yaml +++ b/training/conf/config.yaml @@ -7,8 +7,8 @@ defaults: - hydra: default - logger: wandb - lr_schedulers: - - one_cycle - - mapping: word_piece + - cosine_annealing + - mapping: characters # word_piece - model: lit_transformer - network: conv_transformer - optimizers: @@ -21,6 +21,12 @@ train: true test: true logging: INFO +# datamodule: +# word_pieces: false + +# model: +# max_output_len: 682 + # path to original working directory # hydra hijacks working directory by changing it to the current log directory, # so it's useful to have this path as a special variable |