artelabsuper commited on
Commit
9ebe7d2
1 Parent(s): e3bb30a

add backbone to avoide google download

Browse files
Files changed (3) hide show
  1. .gitattributes +1 -0
  2. mit_b4.pth +3 -0
  3. 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 = './models/weights/mit_b4.pth'
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: