feat: revert GMM config for now

experiments/config/many_well.yaml

@@ -34,9 +34,9 @@ training:
   checkpoint_load_dir: null # or null
   seed: 0
   lr: 3e-4
-  batch_size: 100
+  batch_size: 2048
   n_iterations: null
-  n_flow_forward_pass: 10_000_000_0
+  n_flow_forward_pass: 10_000_000_000
   use_gpu: true
   use_64_bit: true
   use_buffer: false
@@ -53,18 +53,13 @@ training:
 evaluation:
   n_plots: 50 # number of times we visualise the model throughout training.
   n_eval: 50 # for calculating metrics of flow w.r.t target.
-  eval_batch_size: 1000 # must be a multiple of inner batch size
+  eval_batch_size: 2048 # must be a multiple of inner batch size
   n_checkpoints: 10 # number of model checkpoints saved
   save_path:  ./results/many_well/seed${training.seed}/
 
 
 
 logger:
-  wandb:
-    name: Manywell
-    project: fab
-    entity: flow-ais-bootstrap
-    tags: [many_well]
-#  pandas_logger:
-#    save_period: 1000 # how often to save the pandas dataframe as a csv
+  pandas_logger:
+    save_period: 1000 # how often to save the pandas dataframe as a csv