danifei commited on
Commit
86fee05
·
verified ·
1 Parent(s): cf33599

change path of model weights

Browse files
Files changed (1) hide show
  1. app.py +1 -5
app.py CHANGED
@@ -18,12 +18,8 @@ opt = parse(path_opt)
18
 
19
  #define some auxiliary functions
20
  pil_to_tensor = transforms.ToTensor()
21
-
22
- # define some parameters based on the run we want to make
23
- #selected network
24
- network = opt['network']['name']
25
 
26
- PATH_MODEL = opt['save']['path']
27
 
28
  model = Network(img_channel=opt['network']['img_channels'],
29
  width=opt['network']['width'],
 
18
 
19
  #define some auxiliary functions
20
  pil_to_tensor = transforms.ToTensor()
 
 
 
 
21
 
22
+ # PATH_MODEL = opt['save']['best']
23
 
24
  model = Network(img_channel=opt['network']['img_channels'],
25
  width=opt['network']['width'],