def parse_args():
parser = ArgParser(allow_config=True)
parser.add("--ckpt_path", type=path, required=True)
parser.add("--feats_path", type=path, action="append", required=True)
parser.add("--feats_vocab", type=path, action="append", required=True)
parser.add("--labels_vocab", type=path, required=True)
parser.add("--sent_tags", action="store_true", default=False)
parser.add("--save_dir", type=path, required=True)
parser.add("--batch_size", type=int, default=32)
parser.add("--max_length", type=int, default=1e10)
parser.add("--gpu", action="store_true", default=False)
group = parser.add_group("Model Parameters")
group.add("--word_dim", type=int, action="append", required=True)
group.add("--hidden_dim", type=int, required=True)
args = parser.parse_args()
return args
def parse_args():
parser = ArgParser(allow_config=True)
parser.add("--name", type=str, default="main")
parser.add("--feats_path", type=path, action="append", required=True)
parser.add("--feats_vocab", type=path, action="append", required=True)
parser.add("--labels_path", type=path, required=True)
parser.add("--labels_vocab", type=path, required=True)
parser.add("--save_dir", type=path, required=True)
parser.add("--gpu", action="store_true", default=False)
parser.add("--n_previews", type=int, default=10)
group = parser.add_group("Word Embedding Options")
group.add("--wordembed_type",
type=str,
action="append",
choices=["glove", "fasttext", "none"])
group.add("--wordembed_path", type=path, action="append")
group.add("--fasttext_path", type=path, default=None)
group.add("--wordembed_freeze", type=bool, action="append")
group = parser.add_group("Training Options")
group.add("--n_epochs", type=int, default=3)
group.add("--dropout_prob", type=float, default=0.05)
group.add("--batch_size", type=int, default=32)
group.add("--max_len", type=int, default=30)
group = parser.add_group("Save Options")
group.add("--save", action="store_true", default=False)
group.add("--save_period", type=int, default=1000)
group = parser.add_group("Validation Options")
group.add("--val", action="store_true", default=False)
group.add("--val_period", type=int, default=100)
group.add("--text_preview", action="store_true", default=False)
group.add("--val_feats_path", type=path, action="append")
group.add("--val_labels_path", type=path, default=None)
group = parser.add_group("Visdom Options")
group.add("--visdom_host", type=str, default="localhost")
group.add("--visdom_port", type=int, default=8097)
group.add("--visdom_buffer_size", type=int, default=10)
group = parser.add_group("Model Parameters")
group.add("--word_dim", type=int, action="append")
group.add("--hidden_dim", type=int, required=True)
args = parser.parse_args()
return args
