deupdepfaenet configs

RolnickLab · May 17, 2024 · 175567e · 175567e
1 parent 5061d09
commit 175567e
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/configs/exps/deup/uncertainty/deup_depfaenet.yaml b/configs/exps/deup/uncertainty/deup_depfaenet.yaml
@@ -5,7 +5,7 @@ job:
   partition: long
 
 default:
-  config: deup_faenet-deup_is2re-all
+  config: deup_depfaenet-deup_is2re-all
   wandb_project: ocp-deup
   wandb_tags: deup-depfaenet, 4648581-model, 4657270-dataset
   test_ri: True
@@ -16,8 +16,8 @@ default:
     dropout_lin: 0.3
   cp_data_to_tmpdir: false
   inference_time_loops: 1
-  restart_from_dir: /network/scratch/a/alexandre.duval/ocp/runs/4648581/
-  checkpoint: /network/scratch/a/alexandre.duval/ocp/runs/4648581/
+  # restart_from_dir: /network/scratch/a/alexandre.duval/ocp/runs/4648581/
+  # checkpoint: /network/scratch/a/alexandre.duval/ocp/runs/4648581/
   dataset: # mandatory if restart_from_dir is set
     default_val: deup-val_ood_cat-val_ood_ads
     deup-train-val_id:

diff --git a/configs/models/deup_depfaenet.yaml b/configs/models/deup_depfaenet.yaml
@@ -39,7 +39,7 @@ default:
       res_updown:
         hidden_channels: 128
         norm: batch1d # batch1d, layer or null
-    deup_features: [s, energy_pred_std]
+    deup_features: [s, energy_pred_std]  # add q for density
   optim:
     batch_size: 256
     eval_batch_size: 256

diff --git a/configs/models/deup_faenet.yaml b/configs/models/deup_faenet.yaml
@@ -39,7 +39,7 @@ default:
       res_updown:
         hidden_channels: 128
         norm: batch1d # batch1d, layer or null
-    deup_features: [s, energy_pred_std]
+    deup_features: [s, energy_pred_std]  # add q for density if it exists
   optim:
     batch_size: 256
     eval_batch_size: 256