模型训练的时候只保存一个模型

shutil.rmtree(params.save)   #删除文件保存相关文件
os.mkdir(params.save)        #创建相关文件
torch.save(model, params.save+'/model.pkl')  #以torch为例子，然后保存

 

def delete_file(path):
    """
        删除一个目录下的所有文件
    :param path: str, dir path
    :return: None
    """
    for i in os.listdir(path):
        # 取文件或者目录的绝对路径
        path_children = os.path.join(path, i)
        if os.path.isfile(path_children):
            if path_children.endswith(".h5") or path_children.endswith(".json"):
                os.remove(path_children)
        else:# 递归, 删除目录下的所有文件
            delete_file(path_children)