diff --git a/nolearn/lasagne.py b/nolearn/lasagne.py index 35ee029..3618c71 100644 --- a/nolearn/lasagne.py +++ b/nolearn/lasagne.py @@ -129,6 +129,14 @@ def __init__( "Use 'batch_iterator_train' and 'batch_iterator_test' instead." ) + def __getstate__(self): + ret = dict(self.__dict__) + del ret['train_iter_'] + del ret['eval_iter_'] + del ret['predict_iter_'] + + return ret + def _initialize(self): if (getattr(self, 'train_iter_', None) is not None and getattr(self, 'eval_iter_', None) is not None