sfqcyy
/
mindspore
forked from OSSInnovation/mindspore
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Fix permission problem when training with common user.

This commit is contained in:
zhangdengcheng 2020-07-08 08:58:16 +00:00
parent 82531ba413
commit 44aab10bd5
2 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
model_zoo/gat/README.md
View File

@ -72,9 +72,9 @@ sh run_process_data.sh [SRC_PATH] [DATASET_NAME]
>> Launch >> Launch
``` ```
#Generate dataset in mindrecord format for cora #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 #Generate dataset in mindrecord format for citeseer
sh run_process_data.sh citeseer ./run_process_data.sh ./data citeseer
``` ```
# Features # Features

2
model_zoo/gat/train.py
View File

@ -96,6 +96,8 @@ def train():
if eval_acc >= val_acc_max and eval_loss < val_loss_min: if eval_acc >= val_acc_max and eval_loss < val_loss_min:
val_acc_model = eval_acc val_acc_model = eval_acc
val_loss_model = eval_loss 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") _exec_save_checkpoint(train_net.network, "ckpts/gat.ckpt")
val_acc_max = np.max((val_acc_max, eval_acc)) val_acc_max = np.max((val_acc_max, eval_acc))
val_loss_min = np.min((val_loss_min, eval_loss)) val_loss_min = np.min((val_loss_min, eval_loss))