fix tests

Acellera · May 17, 2024 · abc1b1b · abc1b1b
1 parent 7a04a47
commit abc1b1b
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/scripts/pretrain/config.yaml b/scripts/pretrain/config.yaml
@@ -3,18 +3,18 @@ experiment_name: acegen
 agent_name: pretrain_gru
 logger_backend: null # csv, wandb, tensorboard, or null
 seed: 101
-recompute_dataset: True # if False and dataset_log_dir contains a dataset, it will be used
-dataset_log_dir: /tmp/pretrain
-model_log_dir: /tmp/pretrain
 log_frequency: 500
 
 # Dataset configuration
 train_dataset_path: null
-tokenizer: SMILESTokenizer # SMILESTokenizer, SMILESTokenizer2, DeepSMILESTokenizer, SELFIESTokenizer, AISTokenizer, SAFETokenizer, SmiZipTokenizer 
+tokenizer: SMILESTokenizerChEMBL # SMILESTokenizerChEMBL, SMILESTokenizerEnamine, DeepSMILESTokenizer, SELFIESTokenizer, AISTokenizer, SAFETokenizer, SmiZipTokenizer 
+recompute_dataset: True # if False and dataset_log_dir contains a dataset, it will be used
+dataset_log_dir: /tmp/pretrain # if recomputing dataset, save it here
 
 # Model configuration
 model: gru # gru, lstm, or gpt2
 custom_model_factory: null # Path to a custom model factory (e.g. my_module.create_model)
+model_log_dir: /tmp/pretrain # save model here
 
 # Training configuration
 lr: 0.0001