diff --git a/PaddleNLP/PaddleTextGEN/variational_seq2seq/args.py b/PaddleNLP/PaddleTextGEN/variational_seq2seq/args.py index 1d856577490bc5b6dcfbaf0a93b7cb5832a69a0a..a19f65cbce28ce49356d2422da4c888572869111 100644 --- a/PaddleNLP/PaddleTextGEN/variational_seq2seq/args.py +++ b/PaddleNLP/PaddleTextGEN/variational_seq2seq/args.py @@ -117,7 +117,7 @@ def parse_args(): parser.add_argument( '--use_gpu', - type=bool, + type=eval, default=False, help='Whether using gpu [True|False]')