add max_length

HessTaha · May 31, 2020 · c1a3377 · c1a3377
1 parent d03fe60
commit c1a3377
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/main.py b/src/main.py
@@ -56,7 +56,8 @@ def main(path_to_data: str,
 
     config_dict = {
         "vocab_size" : vocab_size,
-        "n_classes"  : n_classes
+        "n_classes"  : n_classes,
+        "max_length" : optimal_length
     }
 
     if n_classes > 2: