prekshyam commited on
Commit
7e3569f
·
verified ·
1 Parent(s): 3afb922

fixed bugs

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -27,7 +27,7 @@ mae_model = MAEViT(
27
  encoder_layers=2, #originally 2
28
  encoder_heads=4, #originally 4
29
  mlp_ratio=2.0, #originally 2.0
30
- mask_ratio=0.60,
31
  decoder_embed_dim=64, #originally 64
32
  decoder_layers=2, #originally 2
33
  decoder_heads=4, #originally 4
@@ -65,7 +65,7 @@ yolo_mapping = {
65
  8: "Surprised"
66
  }
67
 
68
- def mae_reconstruct(image:Image, figure_name='figure/demo_temp.png', mask_ratio):
69
  img = transform(image).unsqueeze(0)
70
  img = img.to(device)
71
  mae_model.mask_ratio = mask_ratio
 
27
  encoder_layers=2, #originally 2
28
  encoder_heads=4, #originally 4
29
  mlp_ratio=2.0, #originally 2.0
30
+ mask_ratio=0.75,
31
  decoder_embed_dim=64, #originally 64
32
  decoder_layers=2, #originally 2
33
  decoder_heads=4, #originally 4
 
65
  8: "Surprised"
66
  }
67
 
68
+ def mae_reconstruct(image:Image, figure_name='figure/demo_temp.png', mask_ratio=0.75):
69
  img = transform(image).unsqueeze(0)
70
  img = img.to(device)
71
  mae_model.mask_ratio = mask_ratio