UWCNN
UWCNN copied to clipboard
Changing Path of Checkpoints
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)