Tiger-J commited on
Commit
80cc1df
·
verified ·
1 Parent(s): 055b8d1

Update app_util.py

Browse files
Files changed (1) hide show
  1. 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))