update demo

r2d3io · Jan 29, 2022 · a176f1e · a176f1e
1 parent 7e37606
commit a176f1e
Show file tree

Hide file tree

Showing 2 changed files with 52 additions and 6 deletions.
diff --git a/demo.ipynb b/demo.ipynb
diff --git a/models/blip.py b/models/blip.py
@@ -23,7 +23,7 @@
 class BLIP_Base(nn.Module):
     def __init__(self,                 
                  med_config = 'configs/med_config.json',  
-                 image_size = 384,
+                 image_size = 224,
                  vit = 'base',
                  vit_grad_ckpt = False,
                  vit_ckpt_layer = 0,