File size: 906 Bytes
3a2cff5 108b713 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 | import torch
# Number of features in each aroma vector
AROMA_VEC_LENGTH: int = 138
# All images are normalized to this
IMG_DIM: int = 224
# Each model was trained with these hyperparams
BATCH_SIZE: int = 16
EMBED_DIM: int = 512
# CPU or GPU?
DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
# Paths to models
OVLE_SMALL_BASE_PATH: str = "./model/colip-small/base/gnn.pth"
ENCODER_SMALL_BASE_PATH: str = "./model/colip-small/base/olf_encoder.pth"
OVLE_LARGE_BASE_PATH: str = "./model/colip-large/base/gnn.pth"
ENCODER_LARGE_BASE_PATH: str = "./model/colip-large/base/olf_encoder.pth"
OVLE_SMALL_GRAPH_PATH: str = "./model/colip-small/graph/gat_gnn.pth"
ENCODER_SMALL_GRAPH_PATH: str = "./model/colip-small/graph/gat_olf_encoder.pth"
OVLE_LARGE_GRAPH_PATH: str = "./model/colip-large/graph/gat_gnn.pth"
ENCODER_LARGE_GRAPH_PATH: str = "./model/colip-large/graph/gat_olf_encoder.pth"
|