unionpoint commited on
Commit
0177bd1
·
verified ·
1 Parent(s): 2a6fc3d

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -0
README.md CHANGED
@@ -38,6 +38,8 @@ model = timm.create_model(
38
  "convnextv2_tiny.fcmae_ft_in22k_in1k_384",
39
  pretrained=False,
40
  num_classes=39,
 
 
41
  )
42
 
43
  # load weights
 
38
  "convnextv2_tiny.fcmae_ft_in22k_in1k_384",
39
  pretrained=False,
40
  num_classes=39,
41
+ drop_rate=0.2,
42
+ drop_path_rate=0.2,
43
  )
44
 
45
  # load weights