From 9c3a0edbe6c9b0e2d792e8edfba849175daefe0f Mon Sep 17 00:00:00 2001
From: Peng Wu <alexepico@gmail.com>
Date: Tue, 26 Jul 2011 14:19:29 +0800
Subject: improve tryprune

---
 tryprune.py | 4 ++++
 1 file changed, 4 insertions(+)

(limited to 'tryprune.py')

diff --git a/tryprune.py b/tryprune.py
index 41a45e9..7a14b0a 100755
--- a/tryprune.py
+++ b/tryprune.py
@@ -198,6 +198,10 @@ if __name__ == '__main__':
     inter_model = os.path.join(trydir, config.getFinalModelFileName())
     convertModel(kmm_model, inter_model)
 
+    modelsize = utils.get_file_length(inter_model)
+    cwdstatus['PruneModelSize'] = modelsize
+    print('final model size:' + modelsize)
+
     #sign status epoch
     utils.sign_epoch(cwdstatus, 'Prune')
     utils.store_status(cwdstatuspath, cwdstatus)
-- 
cgit