deep-learning-for-image-processing _IncompatibleKeys(missing_keys=['pre_logits.fc.weight', 'pre_logits.fc.bias', 'head.weight', 'head.bias'], unexpected

_IncompatibleKeys(missing_keys=['pre_logits.fc.weight', 'pre_logits.fc.bias', 'head.weight', 'head.bias'], unexpected_keys=[])

Open zhangwancheng18 opened this issue 5 months ago • 0 comments

parser.add_argument('--weights', type=str, default='./vit_base_patch16_224.pth',
                    help='initial weights path')
model = create_model(num_classes=args.num_classes, has_logits=True).to(device)

把False改成了Ture就出现了这个警告

Sep 24 '24 14:09 zhangwancheng18

deep-learning-for-image-processing deep-learning-for-image-processing copied to clipboard

_IncompatibleKeys(missing_keys=['pre_logits.fc.weight', 'pre_logits.fc.bias', 'head.weight', 'head.bias'], unexpected_keys=[])

deep-learning-for-image-processing
deep-learning-for-image-processing copied to clipboard