Open fzu021800901 opened 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, )