summaryrefslogtreecommitdiffstats
path: root/tryprune.py
diff options
context:
space:
mode:
Diffstat (limited to 'tryprune.py')
-rwxr-xr-xtryprune.py4
1 files changed, 4 insertions, 0 deletions
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)