File size: 600 Bytes
d60982d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
# environment variables for model training
# environment variables for the inference api
export DATA_DIR=queries
export PATCHES_DIR=${DATA_DIR}/patches
export SLIDES_DIR=${DATA_DIR}/slides
export GRAPHCAM_DIR=${DATA_DIR}/graphcam_plots
mkdir $GRAPHCAM_DIR -p
# manually put the metadata in the metadata folder
export CLASS_METADATA='metadata/label_map.pkl'
# manually put the desired weights in the weights folder
export WEIGHTS_PATH='weights'
export FEATURE_EXTRACTOR_WEIGHT_PATH=${WEIGHTS_PATH}/feature_extractor/model.pth
export GT_WEIGHT_PATH=${WEIGHTS_PATH}/graph_transformer/GraphCAM.pth |