Skip to content

Commit 2ef1af5

Browse files
committed
remove old param
1 parent be1d9f3 commit 2ef1af5

File tree

1 file changed

+0
-5
lines changed

1 file changed

+0
-5
lines changed

references/optical_flow/train.py

Lines changed: 0 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -193,8 +193,6 @@ def main(args):
193193

194194
if args.resume is not None:
195195
d = torch.load(args.resume, map_location="cpu")
196-
if args.map_orig_to_ours:
197-
d = utils.map_orig_to_ours(d)
198196
model.load_state_dict(d, strict=True)
199197

200198
if args.train_dataset is None:
@@ -321,9 +319,6 @@ def get_args_parser(add_help=True):
321319

322320
parser.add_argument("--dist-url", default="env://", help="URL used to set up distributed training")
323321

324-
# TODO: remove
325-
parser.add_argument("--map-orig-to-ours", action="store_true")
326-
327322
parser.add_argument(
328323
"--dataset-root",
329324
help="Root folder where the datasets are stored. Will be passed as the 'root' parameter of the datasets.",

0 commit comments

Comments
 (0)