Update utils

This commit is contained in:
Shangtong Zhang
2018-05-01 13:08:34 -06:00
parent 94adf0e9a1
commit 205fc1d07c
2 changed files with 8 additions and 1 deletions
+4
View File
@@ -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])
+4 -1
View File
@@ -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