| # 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 |