Add beam search

Noble-Lab · Nov 2, 2022 · 0710ee8 · 0710ee8
1 parent a5c05b8
commit 0710ee8
Show file tree

Hide file tree

Showing 4 changed files with 376 additions and 7 deletions.
diff --git a/casanovo/casanovo.py b/casanovo/casanovo.py
@@ -150,6 +150,7 @@ def main(
         model_save_folder_path=str,
         save_weights_only=bool,
         every_n_train_steps=int,
+        n_beams=int,
     )
     for k, t in config_types.items():
         try:

diff --git a/casanovo/config.yaml b/casanovo/config.yaml
@@ -65,6 +65,7 @@ weight_decay: 1e-5
 # Training/inference options.
 train_batch_size: 32
 predict_batch_size: 1024
+n_beams: 1
 
 logger:
 max_epochs: 30