Rename experiments/exp_007_aleph_routed_attention/lexical_atlas.py to experiments/exp_007_aleph_routed_attention/3_lexical_atlas.py

experiments/exp_007_aleph_routed_attention/{lexical_atlas.py → 3_lexical_atlas.py}

@@ -61,6 +61,8 @@ import torch
 import torch.nn.functional as F
 from torch import Tensor
 
+from aleph_trigram_lm import statute
+
 DATASET = "AbstractPhil/wordnet-lexical-topology"
 PAD_BYTE = 0x00                       # reserved pad symbol (documented, learned slot)
 
@@ -482,9 +484,9 @@ if __name__ == "__main__":
     ap = argparse.ArgumentParser(description="Full lexical-topology atlas")
     ap.add_argument("--checkpoint", default=None)
     ap.add_argument("--configs", nargs="+",
-                    default=["all"])#"char_eng_unigram", "char_eng_2gram",
-                             #"char_eng_3gram", "char_eng_4gram",
-                             #"char_eng_5gram"])
+                    default=["char_eng_unigram", "char_eng_2gram",
+                             "char_eng_3gram", "char_eng_4gram",
+                             "char_eng_5gram"])
     ap.add_argument("--d-base", type=int, default=48)
     ap.add_argument("--out", default="atlas")
     ap.add_argument("--device",