CASA-Dialogue-Act-Classifier/config.py at main · SumnerLab/CASA-Dialogue-Act-Classifier · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
import torch

config = {
    # data
    "data_dir":"../input/a2g-dialogue-act-dataset/A2G-Dialogue-Act-Dataset/",
    "data_files":{
        "train":"a2g_train.csv",
        "valid":"a2g_valid.csv",
        "test":"a2g_test.csv",
    },
    "max_len":256,
    "batch_size":8,
    "num_workers":4,

    # model
    "model_name":"roberta-base",
    "hidden_size":768,
    "num_classes":18,

    # training
    "save_dir":"../working/",
    "project":"dialogue-act-classification",
    "run_name":"roberta",
    "lr":1e-5,
    "monitor":"val_accuracy",
    "min_delta":0.001,
    "filepath":"../working/{epoch}-{val_accuracy:4f}",
    "precision":32,
    "epochs":20,
    "device":torch.device("cuda" if torch.cuda.is_available() else "cpu"),


}