utilities/config.py

# import the necessary packages
import os

# define the path to the model
MODEL_PATH = "model"
IMAGENETTE_ATTN_PATH = os.path.join(MODEL_PATH, "att_pool")
IMAGENETTE_STEM_PATH = os.path.join(MODEL_PATH, "stem")
IMAGENETTE_TRUNK_PATH = os.path.join(MODEL_PATH, "trunk")