def saveParams(epoch, params=None):
print "EXPORTING MODEL PARAMS...",
if params == None:
params = l.get_all_param_values(NET)
net_filename = MODEL_PATH + "birdCLEF_" + RUN_NAME + "_model_params_epoch_" + str(epoch) + ".pkl"
if not os.path.exists(MODEL_PATH):
os.makedirs(MODEL_PATH)
with open(net_filename, 'w') as f:
pickle.dump(params, f)
print "DONE!"
评论列表
文章目录