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:
|