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

Changing Path of Checkpoints #9

Open
navjotkaurdhoat opened this issue Jan 23, 2022 · 0 comments
Open

Changing Path of Checkpoints #9

navjotkaurdhoat opened this issue Jan 23, 2022 · 0 comments

Comments

@navjotkaurdhoat
Copy link

Choose the needed model (For example, if you need the type 1 model, please put 'model_checkpoint_path: "coarse.model-type1"' on the first line of checkpoint text.If you are using existing checkpoint text then model in the last line will be loaded(coarse.model-typeII))

In this, where exactly should I change the model's checkpoint path to change the e.g. type 1 model? If changing in the below lines then still it picks up coarse.model-typeII. Why? Any help would be great.

def save(self, checkpoint_dir, step):
    model_name = "coarse.model"
    model_dir = "%s_%s" % ("coarse", self.label_height)
    checkpoint_dir = os.path.join(checkpoint_dir, model_dir)
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

1 participant