File size: 648 Bytes
eff3d61 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
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 = f"./model/ovlc/base/gnn_{EMBED_DIM}_c.pt"
ENCODER_SMALL_BASE_PATH: str = f"./model/ovlc/base/olf_encoder_{EMBED_DIM}_c.pt"
OVLE_LARGE_GRAPH_PATH: str = f"./model/ovlc/graph/gat_gnn_{EMBED_DIM}_c.pt"
ENCODER_LARGE_GRAPH_PATH: str = f"./model/ovlc/graph/gat_olf_encoder_{EMBED_DIM}_c.pt"
|