From 972a24e43a174333572100a6ac1cc645200bd02d Mon Sep 17 00:00:00 2001 From: chentingting Date: Thu, 27 Aug 2020 14:23:05 +0800 Subject: [PATCH] fix parser bool --- model_zoo/official/gnn/gcn/train.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/model_zoo/official/gnn/gcn/train.py b/model_zoo/official/gnn/gcn/train.py index a5e8094804..16360a8968 100644 --- a/model_zoo/official/gnn/gcn/train.py +++ b/model_zoo/official/gnn/gcn/train.py @@ -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)