diff --git a/test/configs/unet_pet.yaml b/test/configs/unet_pet.yaml index b15cef65e1d9bdb4830aad736098507b4893fb57..c81d28513fd84198c85b479107c355a890e96e92 100644 --- a/test/configs/unet_pet.yaml +++ b/test/configs/unet_pet.yaml @@ -33,7 +33,7 @@ TEST: TEST_MODEL: "./test/saved_model/unet_pet/final/" TRAIN: MODEL_SAVE_DIR: "./test/saved_model/unet_pet/" - PRETRAINED_MODEL_DIR: "./test/model/unet_coco/" + PRETRAINED_MODEL_DIR: "./test/models/unet_coco/" SNAPSHOT_EPOCH: 10 SOLVER: NUM_EPOCHS: 500