File size: 722 Bytes
3b99abb | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 | setting:
seed: 20000812
os_environ:
CUDA_VISIBLE_DEVICES: 1,2,3,4,5,6,7
MASTER_ADDR: 127.0.0.1
MASTER_PORT: 12316
WORLD_SIZE: 1
NODE_RANK: 0
out_path: ~
dataset_dir: scripts/LMDB/ClinVar
model:
model_py_path: saprot/saprot_mutation_model
load_prev_scheduler: True
kwargs:
config_path: weight/PLMs/esm2_t33_650M_UR50D
load_pretrained: True
log_clinvar: True
log_dir: output/ClinVar/esm2_t33_650M_UR50D
dataset:
dataset_py_path: mutation_zeroshot_dataset
dataloader_kwargs:
batch_size: 16
num_workers: 8
kwargs: {}
Trainer:
log_every_n_steps: 1
devices: 1
accelerator: gpu
num_nodes: 1
strategy:
find_unused_parameters: False
logger: False
|