File size: 525 Bytes
c91cec6 |
1 2 3 4 5 6 7 8 9 |
FUZZY_SCORE = 80
USE_FUZZY = False # this was used at the initial stage of the evaluation only, then manually reviewed as detailed in the report.
EVALUATION_DATA_PATH = 'my_model/results/evaluation_results.xlsx'
MODEL_NAMES = ['13b', '7b'] # LLaMA-2 varients used.
MODEL_CONFIGURATIONS = ['caption+detic', 'caption+yolov5', 'only_caption', 'only_detic', 'only_yolov5'] # Ablation study on multiple model configurations
OPENAI_API_KEY = os.getenv('OPENAI_API_KEY')
GPT4_MAX_TOKENS=100
GPT4_TEMPERATURE = 0.1
GPT4_SEED = 123 |