| PATH = "/workspace/sg666/MDpLM" | |
| TRAIN_DATA = PATH + "/data/membrane/train.csv" | |
| TEST_DATA = PATH + "/data/membrane/test.csv" | |
| VAL_DATA = PATH + "/data/membrane/val.csv" | |
| ESM_MODEL_PATH = "facebook/esm2_t30_150M_UR50D" | |
| MLM_MODEL_PATH = PATH + "/benchmarks/MLM" | |
| CKPT_DIR = PATH + "/benchmarks/MLM/model_ckpts" | |
| ESM_LAYERS = 3 | |
| BATCH_SIZE = 8 | |
| NUM_EPOCHS = 10 | |
| LEARNING_RATE = 5e-3 | |
| MASKING_RATE = 0.40 |