facebookresearch / ConvNeXt

Code release for ConvNeXt model
MIT License
5.8k stars 697 forks source link

Error:0.3.2timm.create_model did not have layer_scale_init_value #147

Open fzu021800901 opened 1 year ago

fzu021800901 commented 1 year ago
model = create_model(
    args.model, 
    pretrained=False, 
    num_classes=args.nb_classes, 
    drop_path_rate=args.drop_path,
    **layer_scale_init_value=args.layer_scale_init_value,**
    head_init_scale=args.head_init_scale,
    )