Delete spurs_multi
Browse files
spurs_multi/.hydra/config.yaml
DELETED
|
@@ -1,115 +0,0 @@
|
|
| 1 |
-
datamodule:
|
| 2 |
-
_target_: megascale_multi
|
| 3 |
-
batch_size: 1
|
| 4 |
-
num_workers: 8
|
| 5 |
-
pin_memory: true
|
| 6 |
-
alphabet:
|
| 7 |
-
name: esm
|
| 8 |
-
featurizer: megascale
|
| 9 |
-
trainer:
|
| 10 |
-
_target_: pytorch_lightning.Trainer
|
| 11 |
-
gpus: auto
|
| 12 |
-
min_epochs: 1
|
| 13 |
-
max_epochs: 200
|
| 14 |
-
enable_progress_bar: true
|
| 15 |
-
log_every_n_steps: 10
|
| 16 |
-
resume_from_checkpoint: null
|
| 17 |
-
strategy: ddp_sharded_fbo
|
| 18 |
-
precision: 32
|
| 19 |
-
gradient_clip_val: 0.0
|
| 20 |
-
num_sanity_val_steps: 0
|
| 21 |
-
check_val_every_n_epoch: 1
|
| 22 |
-
reload_dataloaders_every_n_epochs: 100000
|
| 23 |
-
replace_sampler_ddp: false
|
| 24 |
-
max_steps: 200000
|
| 25 |
-
name: megascale/spurs_multi
|
| 26 |
-
model:
|
| 27 |
-
_target_: spurs_multi
|
| 28 |
-
encoder:
|
| 29 |
-
d_model: 128
|
| 30 |
-
n_enc_layers: 3
|
| 31 |
-
n_dec_layers: 3
|
| 32 |
-
tune: true
|
| 33 |
-
use_input_decoding_order: true
|
| 34 |
-
adapter_layer_indices:
|
| 35 |
-
- -3
|
| 36 |
-
mlp:
|
| 37 |
-
flat_dim: 128
|
| 38 |
-
input_dim: 256
|
| 39 |
-
hidden_dim: 1024
|
| 40 |
-
num_layers: 4
|
| 41 |
-
output_dim: 21
|
| 42 |
-
dropout: 0.1
|
| 43 |
-
multi:
|
| 44 |
-
flat_dim: 128
|
| 45 |
-
input_dim: 384
|
| 46 |
-
hidden_dim: 768
|
| 47 |
-
num_layers: 4
|
| 48 |
-
output_dim: 1
|
| 49 |
-
dropout: 0.1
|
| 50 |
-
dropout: 0.5
|
| 51 |
-
agg_type: concate
|
| 52 |
-
agg_dim: 128
|
| 53 |
-
task:
|
| 54 |
-
_target_: stability/megascale
|
| 55 |
-
criterion:
|
| 56 |
-
_target_: spurs.modules.mseloss.MSELoss
|
| 57 |
-
optimizer:
|
| 58 |
-
type: adamw
|
| 59 |
-
_partial_: true
|
| 60 |
-
lr: 0.0001
|
| 61 |
-
betas:
|
| 62 |
-
- 0.9
|
| 63 |
-
- 0.98
|
| 64 |
-
weight_decay: 0.0001
|
| 65 |
-
lr_scheduler:
|
| 66 |
-
type: plateau
|
| 67 |
-
lr: ${train.lr}
|
| 68 |
-
factor: 0.2
|
| 69 |
-
mode: min
|
| 70 |
-
monitor: val/rho_avg_mse
|
| 71 |
-
train:
|
| 72 |
-
seed: 49
|
| 73 |
-
lr: 0.001
|
| 74 |
-
monitor: val/rho_avg_spearman
|
| 75 |
-
mode: max
|
| 76 |
-
patience: 10
|
| 77 |
-
optimized_metric: val/rho_avg_spearman
|
| 78 |
-
experiment: mpnn
|
| 79 |
-
callbacks:
|
| 80 |
-
model_summary:
|
| 81 |
-
_target_: pytorch_lightning.callbacks.RichModelSummary
|
| 82 |
-
max_depth: -1
|
| 83 |
-
model_checkpoint:
|
| 84 |
-
_target_: spurs.utils.callbacks.ModelCheckpoint
|
| 85 |
-
monitor: ${train.monitor}
|
| 86 |
-
mode: ${train.mode}
|
| 87 |
-
save_top_k: 1
|
| 88 |
-
save_last: true
|
| 89 |
-
verbose: true
|
| 90 |
-
dirpath: checkpoints
|
| 91 |
-
filename: step_{global_step}-${train.monitor}_{${train.monitor}:.2f}
|
| 92 |
-
auto_insert_metric_name: false
|
| 93 |
-
early_stopping:
|
| 94 |
-
_target_: pytorch_lightning.callbacks.EarlyStopping
|
| 95 |
-
monitor: ${train.monitor}
|
| 96 |
-
mode: ${train.mode}
|
| 97 |
-
patience: ${train.patience}
|
| 98 |
-
min_delta: 0
|
| 99 |
-
check_on_train_epoch_end: false
|
| 100 |
-
logger:
|
| 101 |
-
tensorboard:
|
| 102 |
-
_target_: pytorch_lightning.loggers.tensorboard.TensorBoardLogger
|
| 103 |
-
save_dir: tensorboard/
|
| 104 |
-
name: null
|
| 105 |
-
version: ${name}
|
| 106 |
-
log_graph: false
|
| 107 |
-
default_hp_metric: true
|
| 108 |
-
prefix: ''
|
| 109 |
-
|
| 110 |
-
paths:
|
| 111 |
-
root_dir: ${oc.env:PROJECT_ROOT}
|
| 112 |
-
data_dir: ${paths.root_dir}/data/
|
| 113 |
-
log_dir: ${paths.root_dir}/logs/${name}
|
| 114 |
-
ckpt_dir: ${paths.log_dir}/checkpoints
|
| 115 |
-
psnetdata: ${paths.root_dir}/PSNet/data
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
spurs_multi/checkpoints/best.ckpt
DELETED
|
@@ -1,3 +0,0 @@
|
|
| 1 |
-
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:018081b900a77958f67355f0d323dfdc3e18d63f61c8764f0b268e7c1e32080c
|
| 3 |
-
size 2697880281
|
|
|
|
|
|
|
|
|
|
|
|
spurs_multi/config_tree.log
DELETED
|
@@ -1,118 +0,0 @@
|
|
| 1 |
-
CONFIG
|
| 2 |
-
βββ datamodule
|
| 3 |
-
β βββ _target_: megascale_multi
|
| 4 |
-
β batch_size: 1
|
| 5 |
-
β num_workers: 8
|
| 6 |
-
β pin_memory: true
|
| 7 |
-
β alphabet:
|
| 8 |
-
β name: esm
|
| 9 |
-
β featurizer: megascale
|
| 10 |
-
β
|
| 11 |
-
βββ task
|
| 12 |
-
β βββ _target_: stability/megascale
|
| 13 |
-
β criterion:
|
| 14 |
-
β _target_: spurs.modules.mseloss.MSELoss
|
| 15 |
-
β optimizer:
|
| 16 |
-
β type: adamw
|
| 17 |
-
β _partial_: true
|
| 18 |
-
β lr: 0.0001
|
| 19 |
-
β betas:
|
| 20 |
-
β - 0.9
|
| 21 |
-
β - 0.98
|
| 22 |
-
β weight_decay: 0.0001
|
| 23 |
-
β lr_scheduler:
|
| 24 |
-
β type: plateau
|
| 25 |
-
β lr: 0.001
|
| 26 |
-
β factor: 0.2
|
| 27 |
-
β mode: min
|
| 28 |
-
β monitor: val/rho_avg_mse
|
| 29 |
-
β
|
| 30 |
-
βββ model
|
| 31 |
-
β βββ _target_: spurs_multi
|
| 32 |
-
β encoder:
|
| 33 |
-
β d_model: 128
|
| 34 |
-
β n_enc_layers: 3
|
| 35 |
-
β n_dec_layers: 3
|
| 36 |
-
β tune: true
|
| 37 |
-
β use_input_decoding_order: true
|
| 38 |
-
β adapter_layer_indices:
|
| 39 |
-
β - -3
|
| 40 |
-
β mlp:
|
| 41 |
-
β flat_dim: 128
|
| 42 |
-
β input_dim: 256
|
| 43 |
-
β hidden_dim: 1024
|
| 44 |
-
β num_layers: 2
|
| 45 |
-
β output_dim: 21
|
| 46 |
-
β dropout: 0.1
|
| 47 |
-
β multi:
|
| 48 |
-
β flat_dim: 128
|
| 49 |
-
β input_dim: 384
|
| 50 |
-
β hidden_dim: 768
|
| 51 |
-
β num_layers: 4
|
| 52 |
-
β output_dim: 1
|
| 53 |
-
β dropout: 0.1
|
| 54 |
-
β dropout: 0.5
|
| 55 |
-
β agg_type: concate
|
| 56 |
-
β agg_dim: 128
|
| 57 |
-
β
|
| 58 |
-
βββ callbacks
|
| 59 |
-
β βββ model_summary:
|
| 60 |
-
β _target_: pytorch_lightning.callbacks.RichModelSummary
|
| 61 |
-
β max_depth: -1
|
| 62 |
-
β model_checkpoint:
|
| 63 |
-
β _target_: spurs.utils.callbacks.ModelCheckpoint
|
| 64 |
-
β monitor: val/rho_avg_spearman
|
| 65 |
-
β mode: max
|
| 66 |
-
β save_top_k: 1
|
| 67 |
-
β save_last: true
|
| 68 |
-
β verbose: true
|
| 69 |
-
β dirpath: checkpoints
|
| 70 |
-
β filename: step_{global_step}-val/rho_avg_spearman_{val/rho_avg_spearman:.2f}
|
| 71 |
-
β auto_insert_metric_name: false
|
| 72 |
-
β early_stopping:
|
| 73 |
-
β _target_: pytorch_lightning.callbacks.EarlyStopping
|
| 74 |
-
β monitor: val/rho_avg_spearman
|
| 75 |
-
β mode: max
|
| 76 |
-
β patience: 10
|
| 77 |
-
β min_delta: 0
|
| 78 |
-
β check_on_train_epoch_end: false
|
| 79 |
-
β
|
| 80 |
-
βββ trainer
|
| 81 |
-
β βββ _target_: pytorch_lightning.Trainer
|
| 82 |
-
β gpus: auto
|
| 83 |
-
β min_epochs: 1
|
| 84 |
-
β max_epochs: 10
|
| 85 |
-
β enable_progress_bar: true
|
| 86 |
-
β log_every_n_steps: 10
|
| 87 |
-
β resume_from_checkpoint: null
|
| 88 |
-
β
|
| 89 |
-
βββ experiment_path
|
| 90 |
-
β βββ /nethome/zli3161/DATA-nash/SPURS/data/checkpoints/spurs_multi
|
| 91 |
-
βββ ckpt_path
|
| 92 |
-
β βββ best.ckpt
|
| 93 |
-
βββ data_split
|
| 94 |
-
β βββ test
|
| 95 |
-
βββ mode
|
| 96 |
-
β βββ predict
|
| 97 |
-
βββ name
|
| 98 |
-
β βββ megascale/spurs_multi
|
| 99 |
-
βββ train
|
| 100 |
-
β βββ seed: 49
|
| 101 |
-
β lr: 0.001
|
| 102 |
-
β monitor: val/rho_avg_spearman
|
| 103 |
-
β mode: max
|
| 104 |
-
β patience: 10
|
| 105 |
-
β
|
| 106 |
-
βββ print_config
|
| 107 |
-
β βββ True
|
| 108 |
-
βββ ignore_warnings
|
| 109 |
-
β βββ True
|
| 110 |
-
βββ seed
|
| 111 |
-
β βββ 42
|
| 112 |
-
βββ paths
|
| 113 |
-
βββ root_dir: /work/ziang/SPURS
|
| 114 |
-
data_dir: /work/ziang/SPURS/data/
|
| 115 |
-
log_dir: /nethome/zli3161/DATA-nash/SPURS/data/checkpoints/spurs_multi
|
| 116 |
-
ckpt_dir: /nethome/zli3161/DATA-nash/SPURS/data/checkpoints/spurs_multi/checkpoints
|
| 117 |
-
psnetdata: /work/ziang/SPURS/PSNet/data
|
| 118 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
spurs_multi/multi_additive_before_train_result_25.9.25.pt
DELETED
|
@@ -1,3 +0,0 @@
|
|
| 1 |
-
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:0e7b742b64a73b2591c8c3e8e8c4bd84ed64f90b2d1cca722f62a2339041416b
|
| 3 |
-
size 1237735
|
|
|
|
|
|
|
|
|
|
|
|
spurs_multi/multi_additive_result_25.9.25.pt
DELETED
|
@@ -1,3 +0,0 @@
|
|
| 1 |
-
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:b651f102e1d9ad7b2dca84d29c50e9e8cdb593d0b9d2e18a334b40beb8f3254e
|
| 3 |
-
size 1237735
|
|
|
|
|
|
|
|
|
|
|
|
spurs_multi/multi_result_25.9.25.pt
DELETED
|
@@ -1,3 +0,0 @@
|
|
| 1 |
-
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:d68f86bb83b1a81923c5583132980ae26ebc9ad8448e7396dc965f5cbbdb4f60
|
| 3 |
-
size 1237735
|
|
|
|
|
|
|
|
|
|
|
|