File size: 275 Bytes
364daa0 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 | import torch
DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
MAX_LEN = 20
EMBED_DIM = 300
HIDDEN_DIM = 256
BATCH_SIZE = 32
LR = 1e-3
EPOCHS = 5
MODEL_PATH = "weights/vqa_model.pth"
VOCAB_PATH = "weights/vocab.pkl"
ANSWER_PATH = "weights/answers.pkl"
|