akhaliq HF staff commited on
Commit
4f1bf8b
1 Parent(s): ee8acd8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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=16, metavar="N", help="dataloader threads"
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=False,
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
- map_locatin="cpu",
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',