本文参考GitHub中 Cannot finetune from checkpoint
今天在使用tensorflow中的ssd+mobileNet,因为内存不够报错后,把batch-size改为1后,再继续训练,会出现这个问题,进过上面的GitHub提示,说说在存放checkpoint时,已经有存在的ckpt文件了,要清空才行。
然后在训练就不会出现这个问题了

在使用TensorFlow的SSD与MobileNet进行训练时,遇到内存不足导致的错误。通过调整batch-size并清除旧的checkpoint文件,解决了Cannot finetune from checkpoint的问题。
本文参考GitHub中 Cannot finetune from checkpoint
今天在使用tensorflow中的ssd+mobileNet,因为内存不够报错后,把batch-size改为1后,再继续训练,会出现这个问题,进过上面的GitHub提示,说说在存放checkpoint时,已经有存在的ckpt文件了,要清空才行。
然后在训练就不会出现这个问题了

您可能感兴趣的与本文相关的镜像
TensorFlow-v2.15
TensorFlow 是由Google Brain 团队开发的开源机器学习框架,广泛应用于深度学习研究和生产环境。 它提供了一个灵活的平台,用于构建和训练各种机器学习模型
5万+
4782
7226

被折叠的 条评论
为什么被折叠?