diff options
| author | Gertjan van den Burg <gertjanvandenburg@gmail.com> | 2019-03-06 22:33:28 -0500 |
|---|---|---|
| committer | Gertjan van den Burg <gertjanvandenburg@gmail.com> | 2019-03-06 22:33:28 -0500 |
| commit | 9898417d2af4019004828805db5f07b776181663 (patch) | |
| tree | b7c91614f0636f2be71a2e47f7dd1ab3eec65170 /gensvm | |
| parent | Make sure travis updates packages if it has them (diff) | |
| download | pygensvm-9898417d2af4019004828805db5f07b776181663.tar.gz pygensvm-9898417d2af4019004828805db5f07b776181663.zip | |
Fix some memory leaks
Diffstat (limited to 'gensvm')
| -rw-r--r-- | gensvm/cython_wrapper/wrapper.pyx | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/gensvm/cython_wrapper/wrapper.pyx b/gensvm/cython_wrapper/wrapper.pyx index f98341b..0858103 100644 --- a/gensvm/cython_wrapper/wrapper.pyx +++ b/gensvm/cython_wrapper/wrapper.pyx @@ -82,6 +82,7 @@ def train_wrap( error_msg = check_model(model) if error_msg: gensvm_free_model(model) + gensvm_free_model(seed_model) free_data(data) error_repl = error_msg.decode('utf-8') raise ValueError(error_repl) @@ -106,6 +107,7 @@ def train_wrap( # free model and data gensvm_free_model(model); + gensvm_free_model(seed_model) free_data(data); return (V, n_SV, iter_count, training_error, fit_status) |