Spaces:
Runtime error
Runtime error
artelabsuper
commited on
Commit
•
9ebe7d2
1
Parent(s):
e3bb30a
add backbone to avoide google download
Browse files- .gitattributes +1 -0
- mit_b4.pth +3 -0
- models/model.py +1 -1
.gitattributes
CHANGED
@@ -26,3 +26,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
|
|
26 |
*.zstandard filter=lfs diff=lfs merge=lfs -text
|
27 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
28 |
best_model.ckpt filter=lfs diff=lfs merge=lfs -text
|
|
|
|
26 |
*.zstandard filter=lfs diff=lfs merge=lfs -text
|
27 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
28 |
best_model.ckpt filter=lfs diff=lfs merge=lfs -text
|
29 |
+
mit_b4.pth filter=lfs diff=lfs merge=lfs -text
|
mit_b4.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:d91bbdd26068cbabcadb3e10db5fd04ee56028fc431da5d2fd5e92c2207de300
|
3 |
+
size 245550061
|
models/model.py
CHANGED
@@ -12,7 +12,7 @@ class GLPDepth(nn.Module):
|
|
12 |
|
13 |
self.encoder = mit_b4()
|
14 |
if is_train:
|
15 |
-
ckpt_path = './
|
16 |
try:
|
17 |
load_checkpoint(self.encoder, ckpt_path, logger=None)
|
18 |
except:
|
|
|
12 |
|
13 |
self.encoder = mit_b4()
|
14 |
if is_train:
|
15 |
+
ckpt_path = './mit_b4.pth'
|
16 |
try:
|
17 |
load_checkpoint(self.encoder, ckpt_path, logger=None)
|
18 |
except:
|