Open livelitoday opened 6 years ago
position:
def save_params(param1, param2, param3, param4): import cPickle # <----- write_file = open('params.pkl', 'wb') cPickle.dump(param1, write_file, -1) cPickle.dump(param2, write_file, -1) cPickle.dump(param3, write_file, -1) cPickle.dump(param4, write_file, -1) write_file.close()
python2 import cPickle python3 import pickle
position:
保存训练参数的函数
def save_params(param1, param2, param3, param4): import cPickle # <----- write_file = open('params.pkl', 'wb') cPickle.dump(param1, write_file, -1) cPickle.dump(param2, write_file, -1) cPickle.dump(param3, write_file, -1) cPickle.dump(param4, write_file, -1) write_file.close()