ibaiGorordo commited on
Commit
7ed6df6
1 Parent(s): ee495ec

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -23,7 +23,7 @@ IEEE Transactions on Circuits and Systems for Video Technology (TCSVT)
23
  # Usage
24
  ```python
25
  from model import LDRN
26
- model = LDRN()
27
  model.load_state_dict(torch.load("lap_depth_nyu.pkl"), strict=False)
28
  model.eval()
29
  ```
 
23
  # Usage
24
  ```python
25
  from model import LDRN
26
+ model = LDRN(max_depth=10.0)
27
  model.load_state_dict(torch.load("lap_depth_nyu.pkl"), strict=False)
28
  model.eval()
29
  ```