Spaces:
Runtime error
Runtime error
Update app_util.py
Browse files- app_util.py +1 -1
app_util.py
CHANGED
@@ -137,7 +137,7 @@ class ContextDetDemo():
|
|
137 |
)
|
138 |
model = model.to(device)
|
139 |
|
140 |
-
checkpoint = torch.load(args.resume, map_location='cpu')
|
141 |
missing_keys, unexpected_keys = model.load_state_dict(checkpoint['model'], strict=False)
|
142 |
if len(missing_keys) > 0:
|
143 |
print('Missing Keys: {}'.format(missing_keys))
|
|
|
137 |
)
|
138 |
model = model.to(device)
|
139 |
|
140 |
+
checkpoint = torch.load(args.resume, map_location='cpu', weights_only=False)
|
141 |
missing_keys, unexpected_keys = model.load_state_dict(checkpoint['model'], strict=False)
|
142 |
if len(missing_keys) > 0:
|
143 |
print('Missing Keys: {}'.format(missing_keys))
|