提交 6ce06b77 编写于 作者: M mindspore-ci-bot 提交者: Gitee

!5329 change parser add_argument type bool to ast.literal_eval

Merge pull request !5329 from chentingting/fix_parser_bool
......@@ -19,6 +19,7 @@ GCN training script.
import time
import argparse
import ast
import numpy as np
from matplotlib import pyplot as plt
......@@ -51,7 +52,7 @@ def train():
parser.add_argument('--train_nodes_num', type=int, default=140, help='Nodes numbers for training')
parser.add_argument('--eval_nodes_num', type=int, default=500, help='Nodes numbers for evaluation')
parser.add_argument('--test_nodes_num', type=int, default=1000, help='Nodes numbers for test')
parser.add_argument('--save_TSNE', type=bool, default=False, help='Whether to save t-SNE graph')
parser.add_argument('--save_TSNE', type=ast.literal_eval, default=False, help='Whether to save t-SNE graph')
args_opt = parser.parse_args()
np.random.seed(args_opt.seed)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册