相关文章推荐
上周自己有使用 Kaggle 平台 运行 项目的需求,查阅了一些文章,结合自己的实践经验写了此文章,将使用 Kaggle 平台 运行 项目并查看下载日志 文件 的过程详细记录了下来,一些注意事项也写了,希望能够帮助到大家。 一、上传项目 第一步:创建notebook 1、点击左上角File 2、点击New Notebook,就可以创建一个新的notebook了。 第二步:上传项目和数据集 (建议将项目和数据集分开上传,因为无法在 kaggle 平台上实时更改... kaggle 的output没办法直接下载 文件 夹,倒是可以下载 文件 ,当你的训练模型很多个 文件 的时候, 一个一个下载太慢了, 所以先将output压缩一下,然后下载就行了 直接在cell中 运行 此代码即可 import os import zipfile import datetime def file2zip(packagePath, zipPath): :param packagePath: 文件 夹路径 :param zipPath: 压缩包路径 :return: torch.save(model.state_dict(),'./model_best.pth') 一般情况下,以上语句 保存 的模型在 kaggle 的output/ kaggle /working 文件 夹下 建议将效果较好的模型下载 保存 ,否则网页休眠之后之前训练的结果就都么得了 我这里使用的模型是经过预训练的resnet50 model = torchvision.models.segmentation.fcn_resnet50(pretrained= False,progress= True 如果不用加载的模型,用自己训练的模型,那会有类似这样的代码 def Generator(): inputs = tf.keras.layers.Input(shape=(256, 256, 3)) down_stack = [ down(64, 4, apply_bn=False), #128*128*64 down(128, 4), #64*64*128 down(256, 4), 如何 保存 训练好的模型 通过tf.saved_model.save(netwok, path) 其中network代表你的模型的实例化,path自己定义路径,(记住path 保存 没有 文件 格式,只需要给出路径,例如“./model\my_model” tf.saved
 
推荐文章