ariG23498's picture
fix: reformat code for imagenette
310a06c
# 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")