# 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")