Upload full config files to support future training.

config/config_big.yaml

@@ -0,0 +1,34 @@
+experiment_name: 'runs/transformer_big'
+
+dataset:
+  src_lang: 'lo'
+  src_tokenizer: 'BPE'
+  src_max_seq_len: 400
+  tgt_lang: 'vi'
+  tgt_tokenizer: 'WordLevel'
+  tgt_max_seq_len: 350
+  train_dataset: 'train_clean.dat'
+  validate_dataset: 'dev_clean.dat'
+  tokenizer_file: "tokenizer_{0}.json"
+  bleu_dataset: 'test2023'
+
+model: # 42688527 parameters
+  d_model: 512
+  num_heads: 8
+  d_ff: 2048
+  dropout_p: 0.3
+  num_encoder_layers: 4
+  num_decoder_layers: 2
+  model_folder: "weights"
+  model_basename: "transformer_"
+  preload: "big"
+  bleu_dataset: 'test2023'
+
+train:
+  lr: 0.0001 # 1e-4
+  batch_size: 16
+  num_epochs: 40
+  label_smoothing: 0.1
+  on_colab: True # are you training on Colab?
+  patience: 100 # (steps)
+  warm_up_steps: 700

config/config_huge.yaml

@@ -0,0 +1,34 @@
+experiment_name: 'runs/transformer_huge'
+
+dataset:
+  src_lang: 'lo'
+  src_tokenizer: 'BPE'
+  src_max_seq_len: 400
+  tgt_lang: 'vi'
+  tgt_tokenizer: 'WordLevel'
+  tgt_max_seq_len: 350
+  train_dataset: 'train_clean.dat'
+  validate_dataset: 'dev_clean.dat'
+  tokenizer_file: "tokenizer_{0}.json"
+  bleu_dataset: 'test2023'
+
+model: # 61604879 parameters
+  d_model: 512
+  num_heads: 8
+  d_ff: 2048
+  dropout_p: 0.15
+  num_encoder_layers: 8
+  num_decoder_layers: 4
+  model_folder: "weights"
+  model_basename: "transformer_"
+  preload: "final"
+  bleu_dataset: 'test2023'
+
+train:
+  lr: 0.001 # 1e-2
+  batch_size: 32
+  num_epochs: 50
+  label_smoothing: 0.1
+  on_colab: False # are you training on Colab?
+  patience: 1
+  warm_up_steps: 200

config/config_small.yaml

@@ -0,0 +1,34 @@
+experiment_name: 'runs/transformer_small'
+
+dataset:
+  src_lang: 'lo'
+  src_tokenizer: 'BPE'
+  src_max_seq_len: 400
+  tgt_lang: 'vi'
+  tgt_tokenizer: 'WordLevel'
+  tgt_max_seq_len: 350
+  train_dataset: 'train_clean.dat'
+  validate_dataset: 'dev_clean.dat'
+  tokenizer_file: "tokenizer_{0}.json"
+  bleu_dataset: 'test2023'
+
+model: # 16629775 parameters
+  d_model: 256
+  num_heads: 8
+  d_ff: 1024
+  dropout_p: 0.3
+  num_encoder_layers: 4
+  num_decoder_layers: 2
+  model_folder: "weights"
+  model_basename: "transformer_"
+  preload: "small"
+  bleu_dataset: 'test2023'
+
+train:
+  lr: 0.0001 # 1e-4
+  batch_size: 16
+  num_epochs: 40
+  label_smoothing: 0.1
+  on_colab: True # are you training on Colab?
+  patience: 100 # (steps)
+  warm_up_steps: 700