val metric logging fix

models/rfd3/configs/callbacks/design_callbacks.yaml

@@ -1,5 +1,6 @@
 defaults:
   - train_logging
+  - metrics_logging
   - _self_
 
 log_learning_rate_callback:

models/rfd3/configs/datasets/design_base_rfd3na.yaml

@@ -9,9 +9,9 @@ defaults:
   - train/rna_monomer_distillation@train
   
   # Customized validation datasets
-  - val/unconditional@val.unconditional
-  - val/unconditional_deep@val.unconditional_deep
-  - val/indexed@val.indexed
+  #- val/unconditional@val.unconditional
+  #- val/unconditional_deep@val.unconditional_deep
+  #- val/indexed@val.indexed
   - val/pseudoknot@val.pseudoknot
 
   # Customized train masks
@@ -81,9 +81,9 @@ global_transform_args:
 
   # Used to create simple boolean flags for downstream conditioning
   meta_conditioning_probabilities:
-    # calculate_NA_SS: 1.0
     p_is_nucleic_ss_example: 0.1
     p_nucleic_ss_show_partial_feats: 0.7
+    calculate_NA_SS: 0.5
     calculate_hbonds: 0.2
     calculate_rasa: 0.6
 

models/rfd3/configs/experiment/rfd3na-ss.yaml

@@ -2,14 +2,13 @@
 # Training configuration for RFD3
 
 defaults:
-  # - /debug/default
-  - /debug/default
+  #- /debug/default
   - override /model: rfd3_base
-  - override /logger: null
+  - override /logger: wandb
   - override /datasets: design_base_rfd3na
   - _self_
 
-name: rfd3na-SScond
+name: rfd3na
 tags: [print-model]
 ckpt_path: null
 
@@ -59,6 +58,8 @@ datasets:
       p_is_nucleic_ss_example: 1.0
       p_nucleic_ss_show_partial_feats: 0.7
       p_canonical_bp_filter: 0.2
+      calculate_NA_SS: 0.3
+
     association_scheme: atom23
     #add_na_pair_features: true
     train_conditions:
@@ -94,4 +95,4 @@ trainer:
   #limit_train_batches: 10
   #limit_val_batches: 1
   validate_every_n_epochs: 5
-  prevalidate: true
+  prevalidate: false