def parse_args():
parser = argparse.ArgumentParser()
parser.add_argument("--in-csv", type=str)
parser.add_argument("--fasttext-model", type=str, default=None)
parser.add_argument("--w2v-model", type=str, default=None)
boolean_flag(parser, "normalize", default=False)
parser.add_argument("--txt-sep", type=str, default=" ")
parser.add_argument("--txt-col", type=str)
parser.add_argument("--out-npy", type=str, dest="out_npy", required=True)
parser.add_argument("--n-workers",
type=int,
dest="n_workers",
help="count of workers for dataloader",
default=4)
parser.add_argument("--batch-size",
type=int,
dest="batch_size",
help="dataloader batch size",
default=128)
parser.add_argument("--verbose",
dest="verbose",
action="store_true",
default=False)
args = parser.parse_args()
return args
def parse_args():
parser = argparse.ArgumentParser()
parser.add_argument("--model-dir", type=str, default=None)
parser.add_argument("--config", type=str, default=None)
parser.add_argument("--seed", type=int, default=42)
parser.add_argument("--resume",
default=None,
type=str,
metavar="PATH",
help="path to latest checkpoint")
parser.add_argument("-j",
"--workers",
default=None,
type=int,
metavar="N",
help="number of data loading workers")
parser.add_argument("-b",
"--batch-size",
default=None,
type=int,
metavar="N",
help="mini-batch size")
boolean_flag(parser, "verbose", default=False)
parser.add_argument("--out-prefix", type=str, default=None)
args, unknown_args = parser.parse_known_args()
return args, unknown_args