Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Given your pretrained model, how can I evaluate it on IU dataset? #3

Open
pierremaged opened this issue Nov 25, 2021 · 1 comment
Open

Comments

@pierremaged
Copy link

No description provided.

@zaaachos
Copy link

zaaachos commented Jun 26, 2022

@pierremaged First of all, you should comment out the code-block on the red rectangles in trainer.py. This will make the model run on evaluation mode
image

Secondly, after downloading the pre-trained model, copy the path location of the downloaded model file, and type in the --resume argument a default value with the path location in string.
something like this:
parser.add_argument('--resume', default='path\to\model\file', type=str, help='whether to resume the training from existing checkpoints.')
You could find this in main.py file.
image

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

No branches or pull requests

2 participants