Commit 5ff2f93a authored by Verena Praher's avatar Verena Praher

change evaluation metric for midlevel experiment; doesn't find val_prauc

parent 6c53d1cb
......@@ -71,22 +71,22 @@ def train_mtgjamendo(hparams, midlevel_chkpt_dir):
exp = Experiment(name='mtg', save_dir=CURR_RUN_PATH)
# mtg_configs()
logger.info(f"Loading model from {midlevel_chkpt_dir}")
model = Network(num_targets=7, dataset='mtgjamendo', on_gpu=USE_GPU, load_from=midlevel_chkpt_dir)
model = Network(model_config, hparams, num_targets=7, dataset='mtgjamendo', on_gpu=USE_GPU, load_from=midlevel_chkpt_dir)
logger.info(f"Loaded model successfully")
early_stop = EarlyStopping(
monitor='val_prauc',
monitor='val_loss',
patience=50,
verbose=True,
mode='max'
mode='min'
)
checkpoint_callback = ModelCheckpoint(
filepath=chkpt_dir,
save_best_only=True,
verbose=True,
monitor='val_prauc',
mode='max'
monitor='val_loss',
mode='min'
)
if USE_GPU:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment