diff --git a/utils/config.py b/utils/config.py index 665fbc9..a1c81cf 100644 --- a/utils/config.py +++ b/utils/config.py @@ -59,3 +59,7 @@ class Config: self.test_interval = 0 self.test_repetitions = 10 self.evaluation_env = None + + def merge(self, config_dict): + for key in config_dict.keys(): + setattr(self, key, config_dict[key]) diff --git a/utils/schedule.py b/utils/schedule.py index 7a191e0..dd48f99 100644 --- a/utils/schedule.py +++ b/utils/schedule.py @@ -12,7 +12,10 @@ class ConstantSchedule: return self.val class LinearSchedule: - def __init__(self, start, end, steps): + def __init__(self, start, end=None, steps=None): + if end is None: + end = start + steps = 1 self.inc = (end - start) / float(steps) self.current = start self.end = end