qq_45915612的博客

原创 EfficientNetV2 Finetune on CIFAR10出现head_1/dense/bias is not found in ckpt

这两个变量，如果多读的话，是应该至少有四个的，也就是efficientnetv2-s/head/dense/kernel、efficientnetv2-s/head/dense/bias、efficientnetv2-s/head_1/dense/kernel、efficientnetv2-s/head_1/dense/bias这样。检查代码：输出下载的ckpt的变量名，想看看究竟有没有，结果是，有head的，没有head_1的。其中的/mnt/data2是我存放efficientv2代码文件夹的路径。