lm_train.py

source

Train a new language model on one CPU or one GPU

usage: lm_train.py [-h] [--config CONFIG] [--config2 CONFIG2]
                   [--config3 CONFIG3] [--ngpu NGPU]
                   [--train-dtype {float16,float32,float64,O0,O1,O2,O3}]
                   [--backend {chainer,pytorch}] --outdir OUTDIR
                   [--debugmode DEBUGMODE] --dict DICT [--seed SEED]
                   [--resume [RESUME]] [--verbose VERBOSE]
                   [--tensorboard-dir [TENSORBOARD_DIR]]
                   [--report-interval-iters REPORT_INTERVAL_ITERS]
                   --train-label TRAIN_LABEL --valid-label VALID_LABEL
                   [--test-label TEST_LABEL] [--dump-hdf5-path DUMP_HDF5_PATH]
                   [--opt OPT] [--sortagrad [SORTAGRAD]]
                   [--batchsize BATCHSIZE] [--accum-grad ACCUM_GRAD]
                   [--epoch EPOCH]
                   [--early-stop-criterion [EARLY_STOP_CRITERION]]
                   [--patience [PATIENCE]] [--schedulers SCHEDULERS]
                   [--gradclip GRADCLIP] [--maxlen MAXLEN]
                   [--model-module MODEL_MODULE]

Named Arguments