update with major refactor

config.yaml

@@ -77,7 +77,7 @@ trainer:
       log_weight_decay: false
   - class_path: lightning.pytorch.callbacks.ModelCheckpoint
     init_args:
-      dirpath: genbio_finetune/logs/xtrimo_benchmark_gbft_reproduction/fluorescence_proteinmoe_16B
+      dirpath: modelgenerator/logs/xtrimo_benchmark_gbft_reproduction/fluorescence_proteinmoe_16B
       filename: best_val:{epoch}-{val_spearman:.3f}
       monitor: val_spearman
       verbose: false
@@ -138,10 +138,10 @@ trainer:
   reload_dataloaders_every_n_epochs: 0
   default_root_dir: null
 model:
-  class_path: genbio_finetune.tasks.SequenceRegression
+  class_path: modelgenerator.tasks.SequenceRegression
   init_args:
     backbone:
-      class_path: genbio_finetune.models.proteinfm
+      class_path: modelgenerator.backbones.proteinfm
       init_args:
         from_scratch: false
         max_length: 512
@@ -163,7 +163,7 @@ model:
           attention_probs_dropout_prob: 0
         model_init_args: null
     adapter:
-      class_path: genbio_finetune.models.MLPPoolAdapter
+      class_path: modelgenerator.adapters.MLPPoolAdapter
       init_args:
         pooling: mean_pooling
         hidden_sizes:
@@ -190,7 +190,7 @@ model:
         differentiable: false
         fused: null
     lr_scheduler:
-      class_path: genbio_finetune.lr_schedulers.CosineWithWarmup
+      class_path: modelgenerator.lr_schedulers.CosineWithWarmup
       init_args:
         warmup_ratio: 0.05
         num_warmup_steps: null
@@ -200,7 +200,7 @@ model:
     strict_loading: true
     reset_optimizer_states: false
 data:
-  class_path: genbio_finetune.data.FluorescencePrediction
+  class_path: modelgenerator.data.FluorescencePrediction
   init_args:
     path: biomap-research/fluorescence_prediction
     normalize: true