umuthopeyildirim
commited on
Commit
•
4e2b782
1
Parent(s):
c19a5d3
Fix model loading issue
Browse files
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'])
|