src/config/config.py

from transformers import BertTokenizer
import torch


configuration = {
    "sequence_length": 100,
    "tokenizer_name": "bert-base-uncased",
    "vocab_size": len(BertTokenizer.from_pretrained("bert-base-uncased")),
    "embedding_dim": 10,
    "input_length": 100,
    "num_filters": 128,
    "kernel_size": 4,
    "num_gated_units": 64,
    "hidden_neurons": 128,
    "dropout_cnn": 0.2,
    "dropout_fc": 0.5,
    "device": "cpu"
}