save the model

Hello,
Great implementation, thanks for sharing! I'd like to save the trained model. Do you have an idea of the simplest way to do this? I have tried with pickle but got an error:
TypeError: can't pickle _thread.lock objects

Best,
Eliott