Spaces:
No application file
No application file
File size: 490 Bytes
8cb4f3b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 |
# DESIGNATE constants values for config
DEFAULT_DECODE_STRATEGY = "BeamSearch"
DEFAULT_STRATEGY_KWARGS = {}
DEFAULT_SEED = 101
DEFAULT_BATCH_SIZE = 64
DEFAULT_EVAL_BATCH_SIZE = 8
DEFAULT_TRAIN_TEST_SPLIT = 0.8
DEFAULT_DEVICE = "cpu"
DEFAULT_K = 5
DEFAULT_INPUT_MAX_LENGTH = 200
DEFAULT_MAX_LENGTH = 150
DEFAULT_TRAIN_MAX_LENGTH = 100
DEFAULT_LOWERCASE = True
DEFAULT_NUM_KEEP_MODEL_TRAIN = 5
DEFAULT_NUM_KEEP_MODEL_BEST = 5
DEFAULT_SOS = "<sos>"
DEFAULT_EOS = "<eos>"
DEFAULT_PAD = "<pad>"
|