umuthopeyildirim commited on
Commit
4e2b782
1 Parent(s): c19a5d3

Fix model loading issue

Browse files
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -28,7 +28,6 @@ DEVICE = 'cuda' if torch.cuda.is_available() else 'cpu'
28
  model = NewCRFDepth(version='large07', inv_depth=False,
29
  max_depth=10).to(DEVICE).eval()
30
  model = torch.nn.DataParallel(model)
31
- model.load_state_dict()
32
  checkpoint = torch.load('checkpoints/nyu_L.pth',
33
  map_location=torch.device(DEVICE))
34
  model.load_state_dict(checkpoint['model'])
 
28
  model = NewCRFDepth(version='large07', inv_depth=False,
29
  max_depth=10).to(DEVICE).eval()
30
  model = torch.nn.DataParallel(model)
 
31
  checkpoint = torch.load('checkpoints/nyu_L.pth',
32
  map_location=torch.device(DEVICE))
33
  model.load_state_dict(checkpoint['model'])