Saved trial number in experiment parameters.

1 files changed, 7 insertions, 0 deletions

diff --git a/gen/generator.py b/gen/generator.py
index aed8da3..73ccf45 100644
--- a/gen/generator.py
+++ b/gen/generator.py
@@ -198,9 +198,16 @@ class Generator(object):
 
                 os.mkdir(dir_path)
 
+                if trials > 1:
+                    dp['trial'] = trial
                 self.out_dir = dir_path
+
                 self._create_exp(dict(dp))
+
                 del(self.out_dir)
+                if 'trial' in dp:
+                    del dp['trial']
+
 
     def print_help(self):
         s = str(Template("""Generator $name: