仅作为记录，大佬请跳过。

文章目录

• 背景
• 查看模型

背景

加入创建了一个模型：

model = create_model(
    args.model,
    pretrained=args.pretrained,
    num_classes=2,
    drop_rate=args.drop,
    drop_path_rate=args.drop_path,
    drop_block_rate=args.drop_block,
)

查看模型

lst = model.state_dict().keys()
fc_keys = [k for k in lst]