You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
25 lines
481 B
25 lines
481 B
2 years ago
|
MODEL:
|
||
|
|
BERT_CKPT: "hfl/chinese-macbert-base"
|
||
|
|
DEVICE: "cuda"
|
||
|
|
NAME: "macbert4csc"
|
||
|
|
GPU_IDS: [0]
|
||
|
|
# [loss_coefficient]
|
||
|
|
HYPER_PARAMS: [0.3]
|
||
|
|
#WEIGHTS: "output/macbert4csc/epoch=6-val_loss=0.07.ckpt"
|
||
|
|
WEIGHTS: ""
|
||
|
|
|
||
|
|
DATASETS:
|
||
|
|
TRAIN: "output/train.json"
|
||
|
|
VALID: "output/dev.json"
|
||
|
|
TEST: "output/test.json"
|
||
|
|
|
||
|
|
SOLVER:
|
||
|
|
BASE_LR: 5e-5
|
||
|
|
WEIGHT_DECAY: 0.01
|
||
|
|
BATCH_SIZE: 32
|
||
|
|
MAX_EPOCHS: 10
|
||
|
|
ACCUMULATE_GRAD_BATCHES: 4
|
||
|
|
|
||
|
|
OUTPUT_DIR: "output/macbert4csc"
|
||
|
|
MODE: ["train", "test"]
|