Update app.py
Browse files
app.py
CHANGED
@@ -92,7 +92,7 @@ def load_model():
|
|
92 |
help="dataset name (default: pascal12)",
|
93 |
)
|
94 |
parser.add_argument(
|
95 |
-
"--workers", type=int, default=
|
96 |
)
|
97 |
parser.add_argument(
|
98 |
"--base-size", type=int, default=256, help="base image size"
|
@@ -138,7 +138,7 @@ def load_model():
|
|
138 |
parser.add_argument(
|
139 |
"--no-cuda",
|
140 |
action="store_true",
|
141 |
-
default=
|
142 |
help="disables CUDA training",
|
143 |
)
|
144 |
parser.add_argument(
|
@@ -282,7 +282,7 @@ def load_model():
|
|
282 |
no_batchnorm=False,
|
283 |
widehead=args.widehead,
|
284 |
widehead_hr=args.widehead_hr,
|
285 |
-
|
286 |
arch_option=0,
|
287 |
block_depth=0,
|
288 |
activation='lrelu',
|
|
|
92 |
help="dataset name (default: pascal12)",
|
93 |
)
|
94 |
parser.add_argument(
|
95 |
+
"--workers", type=int, default=1, metavar="N", help="dataloader threads"
|
96 |
)
|
97 |
parser.add_argument(
|
98 |
"--base-size", type=int, default=256, help="base image size"
|
|
|
138 |
parser.add_argument(
|
139 |
"--no-cuda",
|
140 |
action="store_true",
|
141 |
+
default=True,
|
142 |
help="disables CUDA training",
|
143 |
)
|
144 |
parser.add_argument(
|
|
|
282 |
no_batchnorm=False,
|
283 |
widehead=args.widehead,
|
284 |
widehead_hr=args.widehead_hr,
|
285 |
+
map_location="cpu",
|
286 |
arch_option=0,
|
287 |
block_depth=0,
|
288 |
activation='lrelu',
|