Added random config

edorado93 · Jul 9, 2018 · 4f68a84 · 4f68a84
1 parent 0dead58
commit 4f68a84
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/Writing-editing network/configurations.py b/Writing-editing network/configurations.py
@@ -153,6 +153,14 @@ class LargeConfig6(LargeDataset):
     use_topics = True
     experiment_name = "lg-with-topics-lr-0.0001-WE-300"
 
+class RandomConfig(SmallDataset):
+    emsize = 512
+    context_dim = 128
+    lr = 0.0001
+    pretrained = None
+    use_topics = False
+    experiment_name = "random"
+
 configuration = {
                  "st1": SmallTopicsConfig1(),
                  "st2": SmallTopicsConfig2(),
@@ -168,7 +176,8 @@ class LargeConfig6(LargeDataset):
                  "l3": LargeConfig3(),
                  "l4": LargeConfig4(),
                  "l5": LargeConfig5(),
-                 "l6": LargeConfig6()}
+                 "l6": LargeConfig6(),
+                 "random": RandomConfig()}
 
 def get_conf(name):
     return configuration[name]