Spaces:
Runtime error
Runtime error
Bill Psomas
commited on
Commit
·
02a709c
1
Parent(s):
8381087
higher resolution
Browse files
app.py
CHANGED
|
@@ -15,8 +15,13 @@ import vision_transformer as vits
|
|
| 15 |
arch = "vit_small"
|
| 16 |
mode = "simpool"
|
| 17 |
gamma = None
|
|
|
|
| 18 |
patch_size = 16
|
| 19 |
input_size = 224
|
|
|
|
|
|
|
|
|
|
|
|
|
| 20 |
num_classes = 0
|
| 21 |
checkpoint = "checkpoints/vits_dino_simpool_no_gamma_ep100.pth"
|
| 22 |
checkpoint_key = "teacher"
|
|
|
|
| 15 |
arch = "vit_small"
|
| 16 |
mode = "simpool"
|
| 17 |
gamma = None
|
| 18 |
+
|
| 19 |
patch_size = 16
|
| 20 |
input_size = 224
|
| 21 |
+
|
| 22 |
+
patch_size = 16
|
| 23 |
+
input_size = 448
|
| 24 |
+
|
| 25 |
num_classes = 0
|
| 26 |
checkpoint = "checkpoints/vits_dino_simpool_no_gamma_ep100.pth"
|
| 27 |
checkpoint_key = "teacher"
|