Kaggle无网络环境下训练模型后,推理Notebook无法读取模型的问题
Kaggle双Notebook训练推理分离后模型无法访问的解决方案
问题根源
Kaggle中每个Notebook的/kaggle/working是独立的临时工作空间,训练Notebook保存的输出不会自动同步到新创建的推理Notebook中,必须手动配置数据共享。
解决方法
方法一:将模型上传为私有数据集(长期复用推荐)
- 训练阶段:训练完成后,把模型文件夹打包为zip文件,方便后续操作
!zip -r my_model_token.zip /kaggle/working/[你的模型目录名] - 保存训练Notebook版本时,务必勾选Save output,等待版本保存完成。
- 进入训练Notebook的「Versions」页面,找到对应版本,在「Output」区域下载打包好的
my_model_token.zip。 - 登录Kaggle,进入「Datasets」页面,点击「New Dataset」上传该zip文件,创建私有数据集。
- 推理阶段:在推理Notebook中点击右上角「Add data」,搜索并添加自己创建的数据集,然后解压到工作目录:
!unzip /kaggle/input/[你的数据集名称]/my_model_token.zip -d /kaggle/working/
方法二:直接复用训练Notebook的输出版本(快速测试推荐)
- 训练Notebook保存版本并勾选输出后,进入其「Versions」页面,找到目标版本,点击「Use Version」→「New Notebook」。
- 新建的推理Notebook会自动将训练版本的输出挂载到
/kaggle/input/目录下,执行以下命令查看并复制模型到工作目录:!ls /kaggle/input/ !cp -r /kaggle/input/[训练输出路径]/[你的模型目录名] /kaggle/working/
注意事项
- 训练时建议使用绝对路径定义
save_path,比如/kaggle/working/my_model_dir,避免相对路径导致的文件位置混乱。 - 保存训练版本时必须勾选「Save output」,否则所有输出文件(包括模型)不会被持久化保存。
内容的提问来源于stack exchange,提问作者hakim47
相关产品推荐
相关产品推荐

