File size: 645 Bytes
bae913a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
# Which models to benchmark
BENCHMARK_FUSONPLM = True
# FUSONPLM_CKPTS. If you've traiend your own model, this is a dictionary: key = run name, values = epochs
# If you want to use the trained FusOn-pLM, instead FUSONPLM_CKPTS="FusOn-pLM"
FUSONPLM_CKPTS= "FusOn-pLM"
BENCHMARK_ESM = True
# GPU configs
CUDA_VISIBLE_DEVICES="0"
# Overwriting configs
PERMISSION_TO_OVERWRITE_EMBEDDINGS = False # if False, script will halt if it believes these embeddings have already been made.
PERMISSION_TO_OVERWRITE_MODELS = False # if False, script will halt if it believes these embeddings have already been made.
|