Add checkpoints: Thermostability/**

#9
Thermostability/ankh-large/lr5e-4_bt12k_ga8.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"hidden_size": 1536, "num_attention_head": 8, "attention_probs_dropout": 0.1, "plm_model": "ElnaggarLab/ankh-large", "pooling_method": "mean", "pooling_dropout": 0.1, "dataset": "tyang816/Thermostability_ESMFold", "dataset_config": "data/Thermostability/Thermostability_ESMFold_HF.json", "normalize": "min_max", "num_labels": 1, "problem_type": "regression", "pdb_type": "ESMFold", "train_file": null, "valid_file": null, "test_file": null, "metrics": ["spearman_corr"], "seed": 3407, "learning_rate": 0.0005, "scheduler": null, "warmup_steps": 0, "num_workers": 4, "batch_size": null, "batch_token": 12000, "num_epochs": 100, "max_seq_len": -1, "gradient_accumulation_steps": 8, "max_grad_norm": -1, "patience": 10, "monitor": "spearman_corr", "monitor_strategy": "max", "training_method": "freeze", "structure_seq": [], "output_model_name": "lr5e-4_bt12k_ga8.pt", "output_root": "ckpt", "output_dir": "ckpt/Thermostability/ankh-large", "wandb": false, "wandb_entity": null, "wandb_project": "VenusFactory2", "wandb_run_name": null}
Thermostability/ankh-large/lr5e-4_bt12k_ga8.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e7de99c975e73299de4b015ebf5ec302bcb4383e0fc32879062aec8d7a30640a
3
+ size 9464930
Thermostability/esm2_t33_650M_UR50D/lr5e-4_bt12k_ga8.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"hidden_size": 1280, "num_attention_head": 8, "attention_probs_dropout": 0.1, "plm_model": "facebook/esm2_t33_650M_UR50D", "pooling_method": "mean", "pooling_dropout": 0.1, "dataset": "tyang816/Thermostability_ESMFold", "dataset_config": "data/Thermostability/Thermostability_ESMFold_HF.json", "normalize": "min_max", "num_labels": 1, "problem_type": "regression", "pdb_type": "ESMFold", "train_file": null, "valid_file": null, "test_file": null, "metrics": ["spearman_corr"], "seed": 3407, "learning_rate": 0.0005, "scheduler": null, "warmup_steps": 0, "num_workers": 4, "batch_size": null, "batch_token": 12000, "num_epochs": 100, "max_seq_len": -1, "gradient_accumulation_steps": 8, "max_grad_norm": -1, "patience": 10, "monitor": "spearman_corr", "monitor_strategy": "max", "training_method": "freeze", "structure_seq": [], "output_model_name": "lr5e-4_bt12k_ga8.pt", "output_root": "ckpt", "output_dir": "ckpt/Thermostability/esm2_t33_650M_UR50D", "wandb": false, "wandb_entity": null, "wandb_project": "VenusFactory2", "wandb_run_name": null}
Thermostability/esm2_t33_650M_UR50D/lr5e-4_bt12k_ga8.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9fded3217ff4896982f1fba6d2baf69d959ec65d61a36661237831cd52e47c9a
3
+ size 6577250
Thermostability/prot_bert/lr5e-4_bt12k_ga8.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"hidden_size": 1024, "num_attention_head": 8, "attention_probs_dropout": 0.1, "plm_model": "RostLab/prot_bert", "pooling_method": "mean", "pooling_dropout": 0.1, "dataset": "tyang816/Thermostability_ESMFold", "dataset_config": "data/Thermostability/Thermostability_ESMFold_HF.json", "normalize": "min_max", "num_labels": 1, "problem_type": "regression", "pdb_type": "ESMFold", "train_file": null, "valid_file": null, "test_file": null, "metrics": ["spearman_corr"], "seed": 3407, "learning_rate": 0.0005, "scheduler": null, "warmup_steps": 0, "num_workers": 4, "batch_size": null, "batch_token": 12000, "num_epochs": 100, "max_seq_len": -1, "gradient_accumulation_steps": 8, "max_grad_norm": -1, "patience": 10, "monitor": "spearman_corr", "monitor_strategy": "max", "training_method": "freeze", "structure_seq": [], "output_model_name": "lr5e-4_bt12k_ga8.pt", "output_root": "ckpt", "output_dir": "ckpt/Thermostability/prot_bert", "wandb": false, "wandb_entity": null, "wandb_project": "VenusFactory2", "wandb_run_name": null}
Thermostability/prot_bert/lr5e-4_bt12k_ga8.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a9d3fe95a8bd767c94e7d3be8765ce37ea3c9d413d92d9fdcd4866b57b2a64e7
3
+ size 4213858
Thermostability/prot_t5_xl_uniref50/lr5e-4_bt12k_ga8.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"hidden_size": 1024, "num_attention_head": 8, "attention_probs_dropout": 0.1, "plm_model": "RostLab/prot_t5_xl_uniref50", "pooling_method": "mean", "pooling_dropout": 0.1, "dataset": "tyang816/Thermostability_ESMFold", "dataset_config": "data/Thermostability/Thermostability_ESMFold_HF.json", "normalize": "min_max", "num_labels": 1, "problem_type": "regression", "pdb_type": "ESMFold", "train_file": null, "valid_file": null, "test_file": null, "metrics": ["spearman_corr"], "seed": 3407, "learning_rate": 0.0005, "scheduler": null, "warmup_steps": 0, "num_workers": 4, "batch_size": null, "batch_token": 12000, "num_epochs": 100, "max_seq_len": -1, "gradient_accumulation_steps": 8, "max_grad_norm": -1, "patience": 10, "monitor": "spearman_corr", "monitor_strategy": "max", "training_method": "freeze", "structure_seq": [], "output_model_name": "lr5e-4_bt12k_ga8.pt", "output_root": "ckpt", "output_dir": "ckpt/Thermostability/prot_t5_xl_uniref50", "wandb": false, "wandb_entity": null, "wandb_project": "VenusFactory2", "wandb_run_name": null}
Thermostability/prot_t5_xl_uniref50/lr5e-4_bt12k_ga8.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:61ccb7810ef6ec251702d95a4f85fdf2476e50b8ced59e399535ed36cb7a5d50
3
+ size 4213858