danifei commited on
Commit
1623c10
·
verified ·
1 Parent(s): 373e5c5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -29,7 +29,7 @@ model = Network(img_channel=opt['network']['img_channels'],
29
  dilations=opt['network']['dilations'],
30
  extra_depth_wise = opt['network']['extra_depth_wise'])
31
 
32
- checkpoints = torch.load(opt['save']['best'], map_location=device)
33
  # print(checkpoints)
34
  model.load_state_dict(checkpoints['model_state_dict'])
35
 
 
29
  dilations=opt['network']['dilations'],
30
  extra_depth_wise = opt['network']['extra_depth_wise'])
31
 
32
+ checkpoints = torch.load('Network_noFAC_LOLBlur.pt', map_location=device)
33
  # print(checkpoints)
34
  model.load_state_dict(checkpoints['model_state_dict'])
35