convnet.py 文件源码

def cnn_save(self, savepath='cnn.pkl'):
        """Save the trained network

        input
        =====
        savepath: str
            Path of the net to be saved
        """
        import sys
        sys.setrecursionlimit(1000000)
        import pickle
        fp = open(savepath, 'wb')
        # write
        pickle.dump(self.net, fp)
        fp.close()