diff --git a/fluid/PaddleRec/ssr/infer.py b/fluid/PaddleRec/ssr/infer.py index c39669baa75c7e76c1b22ddb1933cdc96808a2e3..d5c9ee1b5dc95eb403932e0ff7534bfadc7568d7 100644 --- a/fluid/PaddleRec/ssr/infer.py +++ b/fluid/PaddleRec/ssr/infer.py @@ -30,7 +30,8 @@ def parse_args(): '--batch_size', type=int, default='50', help='batch_size') parser.add_argument( '--hid_size', type=int, default='128', help='hidden size') - parser.add_argument('--emb_size', type=int, default='128', help='emb size') + parser.add_argument( + '--emb_size', type=int, default='128', help='embedding size') args = parser.parse_args() return args diff --git a/fluid/PaddleRec/ssr/train.py b/fluid/PaddleRec/ssr/train.py index ea10892bf66ae8301ebf732d2a49e6d069e58ae7..8ca5e8ee3b9d79af52a9e75c7540f1e104750b96 100644 --- a/fluid/PaddleRec/ssr/train.py +++ b/fluid/PaddleRec/ssr/train.py @@ -34,10 +34,7 @@ def parse_args(): parser.add_argument( "--base_lr", type=float, default=0.01, help="learning rate") parser.add_argument( - '--vocab_path', - type=str, - default='vocab.txt', - help='vocab file address') + '--vocab_path', type=str, default='vocab.txt', help='vocab file') parser.add_argument( "--epochs", type=int, default=10, help="Number of epochs") parser.add_argument( @@ -50,7 +47,8 @@ def parse_args(): '--model_dir', type=str, default='model_output', help='model dir') parser.add_argument( "--hidden_size", type=int, default=128, help="hidden size") - parser.add_argument("--batch_size", type=int, default=50, help="batch size") + parser.add_argument( + "--batch_size", type=int, default=50, help="number of batch") parser.add_argument( "--embedding_dim", type=int, default=128, help="embedding dim") parser.add_argument(