def main():
parser = argparse.ArgumentParser(fromfile_prefix_chars='@')
parser.add_argument('-edim', type=int, default=128)
parser.add_argument('-hdim', type=int, default=128)
# parser.add_argument('-bsz', type=int, default=32)
parser.add_argument('-bsz', type=int, default=128)
parser.add_argument('-l2reg', type=float, default=0.0)
parser.add_argument('-gpu', type=int, default=-1)
parser.add_argument('-optimizer', type=str, default='adadelta')
parser.add_argument('-lr', type=float, default=1)
parser.add_argument('-nepoches', type=int, default=1000)
# parser.add_argument('-parser_batch', type=int, default=50)
parser.add_argument('-K', type=int, default=15)
parser.add_argument('--halve-lr-every', type=int, default=2)
parser.add_argument('-seed', type=int, default=1000)
parser.add_argument('-entropy_coef', type=float, default=1e-2)
parser.add_argument('-max_length', type=float, default=100)
# parser.add_argument('-load', type=str, default='epoch-734loss-1.2901acc-0.566348.pkl')
parser.add_argument('-load', type=str, default=None)
args = parser.parse_args()
basic.init_seed(args.seed)
# try:
logging.info(f'params:{args}')
train(args)
def main():
parser = argparse.ArgumentParser(fromfile_prefix_chars='@')
parser.add_argument('-load', type=str, default=None)
parser.add_argument('--word-dim', type=int, default=300)
parser.add_argument('--hidden-dim', type=int, default=300)
parser.add_argument('--clf-hidden-dim', type=int, default=1024)
parser.add_argument('--clf-num-layers', type=int, default=1)
parser.add_argument('--leaf-rnn', default=True, action='store_true')
# parser.add_argument('--leaf-rnn', default=False, action='store_true')
# parser.add_argument('--bidirectional', default=True, action='store_true')
parser.add_argument('--bidirectional', default=False, action='store_true')
parser.add_argument('--intra-attention',
default=False,
action='store_true')
parser.add_argument('--batchnorm', default=False, action='store_true')
parser.add_argument('--dropout', default=0.5, type=float)
parser.add_argument('--l2reg', default=0.0, type=float)
parser.add_argument('--pretrained', default='glove.840B.300d')
# parser.add_argument('--pretrained', default=None)
parser.add_argument('--fix-word-embedding',
default=False,
action='store_true')
parser.add_argument('-gpu', default=-1, type=int)
parser.add_argument('--batch-size', type=int, default=64)
parser.add_argument('--max-epoch', type=int, default=1000)
parser.add_argument('--save-dir', type=str, default='.')
parser.add_argument('--omit-prob', default=0.0, type=float)
parser.add_argument('--optimizer', default='adadelta')
parser.add_argument('--fine-grained', default=True, action='store_true')
parser.add_argument('--halve-lr-every', default=2, type=int)
parser.add_argument('--lower', default=False, action='store_true')
parser.add_argument('--latent-tree', default=False, action='store_true')
parser.add_argument('-seed', default=1000, type=int)
# parser.add_argument('-pooling', default='final', type=str)
parser.add_argument('-pooling', default='lstm', type=str)
args = parser.parse_args()
basic.init_seed(args.seed)
logging.info(f'params:{args}')
# try:
train(args)