jichao commited on
Commit
7064790
·
1 Parent(s): 48e1f93

add new model

Browse files
.claude/settings.local.json ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ {
2
+ "permissions": {
3
+ "allow": [
4
+ "Bash(dir:*)"
5
+ ]
6
+ }
7
+ }
.gitattributes CHANGED
@@ -33,3 +33,5 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+
37
+ .claude/
app.py CHANGED
@@ -7,7 +7,7 @@ import numpy as np
7
  import os
8
 
9
  # --- Model Configuration ---
10
- DEFAULT_MODEL_NAME = "mars-ctx-vitb-0217"
11
  MODEL_CONFIGS = {
12
  "mars-ctx-vitb-0217": {
13
  "path": "models/0217-checkpoint-300.pth",
@@ -28,6 +28,12 @@ MODEL_CONFIGS = {
28
  "in_chans": 1,
29
  "description": "ViT-Small/16 (Grayscale Input)"
30
  },
 
 
 
 
 
 
31
  }
32
 
33
  # Global dictionary to store loaded models
 
7
  import os
8
 
9
  # --- Model Configuration ---
10
+ DEFAULT_MODEL_NAME = "dino-vits-mae-100epoch-1217-1220-e50"
11
  MODEL_CONFIGS = {
12
  "mars-ctx-vitb-0217": {
13
  "path": "models/0217-checkpoint-300.pth",
 
28
  "in_chans": 1,
29
  "description": "ViT-Small/16 (Grayscale Input)"
30
  },
31
+ "dino-vits-mae-100epoch-1217-1220-e50": {
32
+ "path": "models/dino-vits-mae-100epoch-1217-1220-e50.pth",
33
+ "timm_id": "vit_small_patch16_224",
34
+ "in_chans": 1,
35
+ "description": "ViT-Small/16 DINO+MAE (Grayscale Input)"
36
+ },
37
  }
38
 
39
  # Global dictionary to store loaded models
models/dino-vits-mae-100epoch-1217-1220-e50.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5a07ee0c6a6787a4cd6ef52b43a4bcfa5efa1499c7ca15a3c03c0e49c11819c5
3
+ size 85933931