smiles_aug_mem, smiles_bar in run.py

wenhao-gao · May 22, 2023 · caa98f3 · caa98f3
1 parent 75fc194
commit caa98f3
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/run.py b/run.py
@@ -63,6 +63,10 @@ def main():
         from main.graph_mcts.run import Graph_MCTS_Optimizer as Optimizer
     elif args.method == 'smiles_ahc':
         from main.smiles_ahc.run import AHC_Optimizer as Optimizer
+    elif args.method == 'smiles_aug_mem':
+        from main.smiles_aug_mem.run import AugmentedMemory_Optimizer as Optimizer 
+    elif args.method == 'smiles_bar':
+        from main.smiles_bar.run import BAR_Optimizer as Optimizer 
     elif args.method == "smiles_lstm_hc":
         from main.smiles_lstm_hc.run import SMILES_LSTM_HC_Optimizer as Optimizer
     elif args.method == 'selfies_lstm_hc':