diff options
author | Peng Wu <alexepico@gmail.com> | 2011-07-26 14:19:29 +0800 |
---|---|---|
committer | Peng Wu <alexepico@gmail.com> | 2011-07-26 14:19:29 +0800 |
commit | 9c3a0edbe6c9b0e2d792e8edfba849175daefe0f (patch) | |
tree | ba604b50c84bc21f6bdfd16574bb55fbfac04f3e /tryprune.py | |
parent | 1966705ee7edc8296e629ee5bb526864a8428faf (diff) | |
download | trainer-9c3a0edbe6c9b0e2d792e8edfba849175daefe0f.tar.gz trainer-9c3a0edbe6c9b0e2d792e8edfba849175daefe0f.tar.xz trainer-9c3a0edbe6c9b0e2d792e8edfba849175daefe0f.zip |
improve tryprune
Diffstat (limited to 'tryprune.py')
-rwxr-xr-x | tryprune.py | 4 |
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) |