我尝试按照“内置图像分类算法入门”教程微调 resnet50 模型。我在我的谷歌云存储中存储了 resnet 50 imagenet 检查点(在此处找到)(路径是 gs://my-bucket/classification/models/resnet-50)。它包含以下内容:
model.ckpt-0090.index
model.ckpt-0090.data-00000-of-00002
model.ckpt-0090.data-00001-of-00002
当我提交训练作业时,我用参数指定pretrained_checkpoint_path
上述路径。作业开始时,我收到以下错误:
ValueError: Couldn't find 'checkpoint' file or checkpoints in given directory gs://my-bucket/classification/models/resnet-50
检查点文件的名称是否应该符合某种预定义的格式?
编辑:实际上,当解压缩时,检查点目录也包含与上面相同的 3 个文件,但前缀为_.
. 我得到了错误,所以我删除了这些前缀文件,但没有更多成功。