Fix typos and path errors on lassen�

LBANN · Mar 5, 2024 · 8d91f8b · 8d91f8b
1 parent 88a0cf7
commit 8d91f8b
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 6 deletions.
diff --git a/applications/FLASK/Transformer/datasets/QM9.py b/applications/FLASK/Transformer/datasets/QM9.py
@@ -17,10 +17,10 @@
 # Load the datasets
 data_dir = os.getenv("QM9_DATA_DIR", "/p/vast1/lbann/datasets/FLASK/QM9")
 
-tokenizer = MolTokenizer("SMILES_vocab.json")
+tokenizer = MolTokenizer(os.path.join(data_dir, "QM9_vocab.json"))
 tokenizer.load_vocab_file()
 
-dataset_train = np.load(os.path.join(data_dir, "QM9_Pretokenized.npy"))
+dataset_train = np.load(os.path.join(data_dir, "QM9_Pretokenized.npy"), allow_pickle=True)
 
 _vocab_size = 46
 

diff --git a/applications/FLASK/Transformer/datasets/pretokenize/SMILES_tokenizer.py b/applications/FLASK/Transformer/datasets/pretokenize/SMILES_tokenizer.py
@@ -39,7 +39,7 @@ def load_vocab_file(self):
             with open(self.vocab_file, "r") as f:
                 self.vocab_dict = json.load(f)
         else:
-            raise NameError("Vocab file not douns")
+            raise NameError(f"Vocab file not found in {self.vocab_file}")
 
     def load_vocab_dict(self, vocab_dict):
         self.vocab_dict = vocab_dict

diff --git a/applications/FLASK/Transformer/network.py b/applications/FLASK/Transformer/network.py
@@ -4,7 +4,7 @@
 import math
 from typing import Tuple
 from lbann.models.transformer import LayerNorm
-from lbann.modules.transformer import PositionalEncoding
+from lbann.modules.transformer.encoding import PositionalEncoding
 import numpy as np
 
 

diff --git a/applications/FLASK/Transformer/trainer.py b/applications/FLASK/Transformer/trainer.py
@@ -16,7 +16,7 @@
 
 import dataset_utils
 import arg_utils
-import applications.FLASK.Transformer.network as network
+import network
 
 
 def construct_training_task(
@@ -276,7 +276,7 @@ def main():
     lbann.contrib.args.add_training_arguments(parser)
     arg_utils.add_transformer_architecture_arguments(parser)
     arg_utils.add_training_arguments(parser)
-    arg_utils.add_dataset_arguments(parser, default="qm9")
+    arg_utils.add_dataset_arguments(parser, default="QM9")
 
     parser.add_argument(
         "--optimizer",