Update.

diff --git a/main.py b/main.py
index c0f9e57..8726c96 100755 (executable)
--- a/main.py
+++ b/main.py
@@ -1266,8 +1266,8 @@ noise_proba = 0.05
 models = []
 
 for i in range(args.nb_models):
-    # model = MyAttentionAE(
-    model = FunctionalAE(
+    model = MyAttentionAE(
+        # model = FunctionalAE(
         vocabulary_size=vocabulary_size,
         dim_model=args.dim_model,
         dim_keys=args.dim_keys,