re-design epochs

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')