source/config.py

import os
import torch


class Config(object):

    def __init__(self) -> None:

        self.DEVICE = torch.device("cpu")

        self.VOCAB_FILE = 'source/vocabulary.txt'
        self.VOCAB_SIZE = 5000

        self.NUM_LAYER = 1
        self.IMAGE_EMB_DIM = 256
        self.WORD_EMB_DIM = 256
        self.HIDDEN_DIM = 512

        self.EMBEDDING_WEIGHT_FILE = 'source/weights/embeddings-32B-512H-1L-e5.pt'
        self.ENCODER_WEIGHT_FILE = 'source/weights/encoder-32B-512H-1L-e5.pt'
        self.DECODER_WEIGHT_FILE = 'source/weights/decoder-32B-512H-1L-e5.pt'

        self.ROOT = os.path.join(os.path.expanduser('~'), 'Huggingface', 'ImageCaption')