UWCNN icon indicating copy to clipboard operation
UWCNN copied to clipboard

Changing Path of Checkpoints

Open navjotkaurdhoat opened this issue 3 years ago • 1 comments

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)

navjotkaurdhoat avatar Jan 23 '22 08:01 navjotkaurdhoat