From 44aab10bd5f92724b78d07b9131113f5602ad598 Mon Sep 17 00:00:00 2001 From: zhangdengcheng Date: Wed, 8 Jul 2020 08:58:16 +0000 Subject: [PATCH] Fix permission problem when training with common user. --- model_zoo/gat/README.md | 4 ++-- model_zoo/gat/train.py | 2 ++ 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/model_zoo/gat/README.md b/model_zoo/gat/README.md index 7c30e08851..0c46aebbaf 100644 --- a/model_zoo/gat/README.md +++ b/model_zoo/gat/README.md @@ -72,9 +72,9 @@ sh run_process_data.sh [SRC_PATH] [DATASET_NAME] >> Launch ``` #Generate dataset in mindrecord format for cora -sh run_process_data.sh cora +./run_process_data.sh ./data cora #Generate dataset in mindrecord format for citeseer -sh run_process_data.sh citeseer +./run_process_data.sh ./data citeseer ``` # Features diff --git a/model_zoo/gat/train.py b/model_zoo/gat/train.py index af1808b995..acfbb05b78 100644 --- a/model_zoo/gat/train.py +++ b/model_zoo/gat/train.py @@ -96,6 +96,8 @@ def train(): if eval_acc >= val_acc_max and eval_loss < val_loss_min: val_acc_model = eval_acc val_loss_model = eval_loss + if os.path.exists("ckpts/gat.ckpt"): + os.remove("ckpts/gat.ckpt") _exec_save_checkpoint(train_net.network, "ckpts/gat.ckpt") val_acc_max = np.max((val_acc_max, eval_acc)) val_loss_min = np.min((val_loss_min, eval_loss))