Spaces:
Running
Running
change path of model weights
Browse files
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']['
|
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'],
|