From 23f7939071216e70223a0853a519697f80128090 Mon Sep 17 00:00:00 2001
From: =?utf8?q?Fran=C3=A7ois=20Fleuret?= <francois@fleuret.org>
Date: Sat, 31 Aug 2024 10:03:05 +0200
Subject: [PATCH] Update.

---
 main.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/main.py b/main.py
index c0f9e57..8726c96 100755
--- 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,
-- 
2.39.5