diff options
Diffstat (limited to 'lib')
-rw-r--r-- | lib/myconfig.py | 16 | ||||
-rw-r--r-- | lib/utils.py | 6 |
2 files changed, 8 insertions, 14 deletions
diff --git a/lib/myconfig.py b/lib/myconfig.py index 5e95582..7f46676 100644 --- a/lib/myconfig.py +++ b/lib/myconfig.py @@ -11,20 +11,8 @@ class MyConfig: 'EvaluateEpoch': 5 \ } - def getSegmentEpoch(self): - return self.m_current_epoch['SegmentEpoch'] - - def getGenerateEpoch(self): - return self.m_current_epoch['GenerateEpoch'] - - def getEstimateEpoch(self): - return self.m_current_epoch['EstimateEpoch'] - - def getPruneEpoch(self): - return self.m_current_epoch['PruneEpoch'] - - def getEvaluateEpoch(self): - return self.m_current_epoch['EvaluateEpoch'] + def getEpochs(self): + return self.m_current_epoch m_trainer_dir = '/media/data/Program/trainer' diff --git a/lib/utils.py b/lib/utils.py index 156fbaf..9024c7e 100644 --- a/lib/utils.py +++ b/lib/utils.py @@ -28,6 +28,12 @@ def store_status(outfile, obj): write_file(outfile, json.dumps(obj)) return +def check_epoch(obj, passname): + pass + +def sign_epoch(obj, passname): + pass + #test case if __name__ == '__main__': obj = load_status('/tmp/test.status') |