From 50fffc0f50791e0058e1e685614de5f7cb24d924 Mon Sep 17 00:00:00 2001 From: jiahuei Date: Mon, 4 Jan 2021 19:18:34 +0800 Subject: [PATCH] Update testing.sh --- testing.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/testing.sh b/testing.sh index 0a8cb5a..5478904 100644 --- a/testing.sh +++ b/testing.sh @@ -104,7 +104,7 @@ python /master/src/caption_vae/train_n_prune_transformer.py \ --save_checkpoint_every 10 \ --cache_min_free_ram ${CACHE_FREE_RAM} -exit 1 + # Fine-tune with mask frozen BASELINE="${LOG_DIR}/${MODEL_ID}__supermask__${PRUNE_SPARSITY_TARGET}__wg_120.0/model_best_bin_mask.pth" EPOCHS=10