We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent be1d9f3 commit 2ef1af5Copy full SHA for 2ef1af5
references/optical_flow/train.py
@@ -193,8 +193,6 @@ def main(args):
193
194
if args.resume is not None:
195
d = torch.load(args.resume, map_location="cpu")
196
- if args.map_orig_to_ours:
197
- d = utils.map_orig_to_ours(d)
198
model.load_state_dict(d, strict=True)
199
200
if args.train_dataset is None:
@@ -321,9 +319,6 @@ def get_args_parser(add_help=True):
321
319
322
320
parser.add_argument("--dist-url", default="env://", help="URL used to set up distributed training")
323
324
- # TODO: remove
325
- parser.add_argument("--map-orig-to-ours", action="store_true")
326
-
327
parser.add_argument(
328
"--dataset-root",
329
help="Root folder where the datasets are stored. Will be passed as the 'root' parameter of the datasets.",
0 commit comments