def main(args): if args.summary_vocab is None: args.summary_vocab = args.summary + "_vocab" if args.summary_label is None: args.summary_label = args.summary + "_label" assert os.path.isfile(args.summary) assert os.path.isfile(args.summary_vocab) assert os.path.isfile(args.summary_label) print_args(args) summary_dico = Dictionary.read_vocab(args.summary_vocab) summary_data = Dictionary.index_summary(args.summary, args.summary_label, summary_dico, args.summary + ".pth", max_len=args.summary_max_length)
if __name__ == '__main__': readme = "" parser = argparse.ArgumentParser(description=readme) parser.add_argument('--summary', help="summary data") parser.add_argument('--summary_vocab', help="summary data vocab") parser.add_argument('--summary_label', help="summary data label") parser.add_argument('--summary_max_length', type=int, default=600, help="summmary maximum length") args = parser.parse_args() if args.summary_vocab is None: args.summary_vocab = args.summary + "_vocab" if args.summary_label is None: args.summary_label = args.summary + "_label" assert os.path.isfile(args.summary) assert os.path.isfile(args.summary_vocab) assert os.path.isfile(args.summary_label) print_args(args) summary_dico = Dictionary.read_vocab(args.summary_vocab) summary_data = Dictionary.index_summary(args.summary, args.summary_label, summary_dico, args.summary + ".pth", max_len=args.summary_max_length)