tyang816 commited on
Commit
f8f4def
1 Parent(s): 35f3be5

Upload 9 files

Browse files
config/k10_h1280.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"noise_ratio": 0.05, "noise_type": "mut", "gnn": "egnn", "gnn_config": {"hidden_channels": 1280, "edge_attr_dim": 93, "output_dim": 20, "dropout": 0, "n_layers": 6, "residual": false, "embedding": false, "embedding_dim": 64, "mlp_num": 2}, "esm_param": "facebook/esm2_t33_650M_UR50D", "node_dim": 26, "edge_dim": 93, "seed": 12345, "learning_rate": 0.0001, "warmup_steps": 660, "warmup_percent": 0.2, "weight_decay": 0.0, "num_train_epochs": 60, "epoch_idx": 0, "auto_continue_train": true, "batch_token_num": 8192, "max_graph_token_num": 2000, "patience": 10, "max_grad_norm": 4.0, "gradient_accumulation_steps": 10, "cath_dataset": "data/cath40_k10", "dataset_K": 10, "portion": 40, "ckpt_path": "result/ckpt/0417_k10_h1280", "n_gpu": 1, "plm_hidden_size": 1280}
config/k10_h512.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"noise_ratio": 0.05, "noise_type": "mut", "gnn": "egnn", "gnn_config": {"hidden_channels": 512, "edge_attr_dim": 93, "output_dim": 20, "dropout": 0, "n_layers": 6, "residual": false, "embedding": false, "embedding_dim": 64, "mlp_num": 2}, "esm_param": "facebook/esm2_t33_650M_UR50D", "node_dim": 26, "edge_dim": 93, "seed": 12345, "learning_rate": 0.0001, "warmup_steps": 660, "warmup_percent": 0.2, "weight_decay": 0.0, "num_train_epochs": 60, "epoch_idx": 0, "auto_continue_train": false, "batch_token_num": 8192, "max_graph_token_num": 2000, "patience": 10, "max_grad_norm": 4.0, "gradient_accumulation_steps": 10, "cath_dataset": "data/cath40_k10", "dataset_K": 10, "portion": 40, "ckpt_path": "result/ckpt/0417_k10_h512", "n_gpu": 1, "plm_hidden_size": 1280}
config/k10_h768.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"noise_ratio": 0.05, "noise_type": "mut", "gnn": "egnn", "gnn_config": {"hidden_channels": 768, "edge_attr_dim": 93, "output_dim": 20, "dropout": 0, "n_layers": 6, "residual": false, "embedding": false, "embedding_dim": 64, "mlp_num": 2}, "esm_param": "facebook/esm2_t33_650M_UR50D", "node_dim": 26, "edge_dim": 93, "seed": 12345, "learning_rate": 0.0001, "warmup_steps": 660, "warmup_percent": 0.2, "weight_decay": 0.0, "num_train_epochs": 60, "epoch_idx": 0, "auto_continue_train": true, "batch_token_num": 8192, "max_graph_token_num": 2000, "patience": 10, "max_grad_norm": 4.0, "gradient_accumulation_steps": 10, "cath_dataset": "data/cath40_k10", "dataset_K": 10, "portion": 40, "ckpt_path": "result/ckpt/0417_k10_h768", "n_gpu": 1, "plm_hidden_size": 1280}
config/k20_h1280.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"noise_ratio": 0.05, "noise_type": "mut", "gnn": "egnn", "gnn_config": {"hidden_channels": 1280, "edge_attr_dim": 93, "output_dim": 20, "dropout": 0, "n_layers": 6, "residual": false, "embedding": false, "embedding_dim": 64, "mlp_num": 2}, "esm_param": "facebook/esm2_t33_650M_UR50D", "node_dim": 26, "edge_dim": 93, "seed": 12345, "learning_rate": 0.0001, "warmup_steps": 696, "warmup_percent": 0.2, "weight_decay": 0.0, "num_train_epochs": 60, "epoch_idx": 0, "auto_continue_train": true, "batch_token_num": 8192, "max_graph_token_num": 2000, "patience": 10, "max_grad_norm": 4.0, "gradient_accumulation_steps": 10, "cath_dataset": "data/cath40_k20", "dataset_K": 20, "portion": 40, "ckpt_path": "result/ckpt/0417_k20_h1280", "n_gpu": 1, "plm_hidden_size": 1280}
config/k20_h512.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"noise_ratio": 0.05, "noise_type": "mut", "gnn": "egnn", "gnn_config": {"hidden_channels": 512, "edge_attr_dim": 93, "output_dim": 20, "dropout": 0, "n_layers": 6, "residual": false, "embedding": false, "embedding_dim": 64, "mlp_num": 2}, "esm_param": "facebook/esm2_t33_650M_UR50D", "node_dim": 26, "edge_dim": 93, "seed": 12345, "learning_rate": 0.0001, "warmup_steps": 684, "warmup_percent": 0.2, "weight_decay": 0.0, "num_train_epochs": 60, "epoch_idx": 0, "auto_continue_train": true, "batch_token_num": 8192, "max_graph_token_num": 2000, "patience": 10, "max_grad_norm": 4.0, "gradient_accumulation_steps": 10, "cath_dataset": "data/cath40_k20", "dataset_K": 20, "portion": 40, "ckpt_path": "result/ckpt/0417_k20_h512", "n_gpu": 1, "plm_hidden_size": 1280}
config/k20_h768.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"noise_ratio": 0.05, "noise_type": "mut", "gnn": "egnn", "gnn_config": {"hidden_channels": 768, "edge_attr_dim": 93, "output_dim": 20, "dropout": 0, "n_layers": 6, "residual": false, "embedding": false, "embedding_dim": 64, "mlp_num": 2}, "esm_param": "facebook/esm2_t33_650M_UR50D", "node_dim": 26, "edge_dim": 93, "seed": 12345, "learning_rate": 0.0001, "warmup_steps": 684, "warmup_percent": 0.2, "weight_decay": 0.0, "num_train_epochs": 60, "epoch_idx": 0, "auto_continue_train": true, "batch_token_num": 8192, "max_graph_token_num": 2000, "patience": 10, "max_grad_norm": 4.0, "gradient_accumulation_steps": 10, "cath_dataset": "data/cath40_k20", "dataset_K": 20, "portion": 40, "ckpt_path": "result/ckpt/0417_k20_h768", "n_gpu": 1, "plm_hidden_size": 1280}
config/k30_h1280.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"noise_ratio": 0.05, "noise_type": "mut", "gnn": "egnn", "gnn_config": {"hidden_channels": 1280, "edge_attr_dim": 93, "output_dim": 20, "dropout": 0, "n_layers": 6, "residual": false, "embedding": false, "embedding_dim": 64, "mlp_num": 2}, "esm_param": "facebook/esm2_t33_650M_UR50D", "node_dim": 26, "edge_dim": 93, "seed": 12345, "learning_rate": 0.0001, "warmup_steps": 1404, "warmup_percent": 0.2, "weight_decay": 0.0, "num_train_epochs": 60, "epoch_idx": 0, "auto_continue_train": true, "batch_token_num": 4096, "max_graph_token_num": 2000, "patience": 10, "max_grad_norm": 4.0, "gradient_accumulation_steps": 10, "cath_dataset": "data/cath40_k30", "dataset_K": 30, "portion": 40, "ckpt_path": "result/ckpt/0417_k30_h1280", "n_gpu": 1, "plm_hidden_size": 1280}
config/k30_h512.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"noise_ratio": 0.05, "noise_type": "mut", "gnn": "egnn", "gnn_config": {"hidden_channels": 512, "edge_attr_dim": 93, "output_dim": 20, "dropout": 0, "n_layers": 6, "residual": false, "embedding": false, "embedding_dim": 64, "mlp_num": 2}, "esm_param": "facebook/esm2_t33_650M_UR50D", "node_dim": 26, "edge_dim": 93, "seed": 12345, "learning_rate": 0.0001, "warmup_steps": 1404, "warmup_percent": 0.2, "weight_decay": 0.0, "num_train_epochs": 60, "epoch_idx": 0, "auto_continue_train": true, "batch_token_num": 4096, "max_graph_token_num": 2000, "patience": 10, "max_grad_norm": 4.0, "gradient_accumulation_steps": 10, "cath_dataset": "data/cath40_k30", "dataset_K": 30, "portion": 40, "ckpt_path": "result/ckpt/0417_k30_h512", "n_gpu": 1, "plm_hidden_size": 1280}
config/k30_h768.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"noise_ratio": 0.05, "noise_type": "mut", "gnn": "egnn", "gnn_config": {"hidden_channels": 768, "edge_attr_dim": 93, "output_dim": 20, "dropout": 0, "n_layers": 6, "residual": false, "embedding": false, "embedding_dim": 64, "mlp_num": 2}, "esm_param": "facebook/esm2_t33_650M_UR50D", "node_dim": 26, "edge_dim": 93, "seed": 12345, "learning_rate": 0.0001, "warmup_steps": 1404, "warmup_percent": 0.2, "weight_decay": 0.0, "num_train_epochs": 60, "epoch_idx": 0, "auto_continue_train": true, "batch_token_num": 4096, "max_graph_token_num": 2000, "patience": 10, "max_grad_norm": 4.0, "gradient_accumulation_steps": 10, "cath_dataset": "data/cath40_k30", "dataset_K": 30, "portion": 40, "ckpt_path": "result/ckpt/0417_k30_h768", "n_gpu": 1, "plm_hidden_size": 1280}