parser.add_argument('--batch_size', type=int, default=256, help='input batch size') parser.add_argument('--embed_dim', type=int, default=64, help='the dimension of ...