File size: 937 Bytes
8d9d9da |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
# Benchmarking configs
BENCHMARK_FUSONPLM = True # True if you want to benchmark a FusOn-pLM Model
# 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"
# Model comparison configs
BENCHMARK_ESM = True # True if you want to benchmark ESM-2-650M
BENCHMARK_PROTT5 = True # True if you want to benchmark ProtT5
BENCHMARK_FO_PUNCTA_ML = True # True if you want to benchmark FO-Puncta-ML from the FOdb paper
# Overwriting configs
PERMISSION_TO_OVERWRITE = False # if False, script will halt if it believes these embeddings have already been made.
# GPU configs
CUDA_VISIBLE_DEVICES="0" # GPUs to make visible for this process |