diff options
Diffstat (limited to 'src/experiment.py')
-rw-r--r-- | src/experiment.py | 9 |
1 files changed, 5 insertions, 4 deletions
diff --git a/src/experiment.py b/src/experiment.py index f8a219a..85bc4a1 100644 --- a/src/experiment.py +++ b/src/experiment.py @@ -9,8 +9,8 @@ import torch from runner import Runner class Experiment: - def __init__(self, file): - self.make_dir() + def __init__(self, file, prefix): + self.make_dir(prefix) self.copy_config(file) self.metrics = ExperimentMetrics() self.runner = Runner(file, self.metrics) @@ -41,9 +41,10 @@ class Experiment: def dir_path(self, file): return '{}/{}'.format(self.dirname, file) - def make_dir(self): + def make_dir(self, prefix): time_string = time.strftime('%Y%m%d%H%M%S') - dirname = 'outputs/{}'.format(time_string) + prefix = '' if prefix == '' else '{}-'.format(prefix) + dirname = 'outputs/{}{}'.format(prefix, time_string) self.dirname = dirname os.mkdir(dirname) |