Spaces:
Sleeping
Sleeping
File size: 572 Bytes
cb7427c 9a90e40 81de229 cb7427c 9f7e6ed cb7427c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
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 = 512
self.WORD_EMB_DIM = 512
self.HIDDEN_DIM = 512
self.EMBEDDING_WEIGHT_FILE = 'source/weights/embeddings-32B-512H-1L-e10.pt'
self.ENCODER_WEIGHT_FILE = 'source/weights/encoder-32B-512H-1L-e10.pt'
self.DECODER_WEIGHT_FILE = 'source/weights/decoder-32B-512H-1L-e10.pt'
|