Skip to content

fix: test_criterion falls back to undefined args.criterion - #107

Open
andrewwhitecdw wants to merge 1 commit into
NVlabs:mainfrom
andrewwhitecdw:bugfix/training-test-criterion-falls-back-to-undefined
Open

fix: test_criterion falls back to undefined args.criterion#107
andrewwhitecdw wants to merge 1 commit into
NVlabs:mainfrom
andrewwhitecdw:bugfix/training-test-criterion-falls-back-to-undefined

Conversation

@andrewwhitecdw

Copy link
Copy Markdown

Problem

fix: test_criterion falls back to undefined args.criterion

Fix

Replace:

    test_criterion = eval(args.test_criterion or args.criterion).to(device)

with:

    test_criterion = eval(args.test_criterion or args.train_criterion).to(device)

Files changed

  • dust3r/training.py

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant