update configs

PIQuIL · Aug 22, 2023 · 0e48d34 · 0e48d34
1 parent 771be4b
commit 0e48d34
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 21 deletions.
diff --git a/config/config_large.yaml b/config/config_large.yaml
@@ -34,18 +34,15 @@ optimizer:
   eta_min: 0.00001
 
 dataset:
-  num_atoms: null
-  num_samples: null
-  delta: null
-  num_workers: 4
+  num_workers: 8
 
 rydberg:
   num_states: 2
-  num_encoder_embedding_dims: 3
 
 profiler: 
   profiler: "PyTorchProfiler"
+  advanced_monitoring: False
 
 misc:
   seed: 8596
-  prog_bar: True
+  prog_bar: False
diff --git a/config/config_medium.yaml b/config/config_medium.yaml
@@ -34,18 +34,15 @@ optimizer:
   eta_min: 0.00001
 
 dataset:
-  num_atoms: null
-  num_samples: null
-  delta: null
-  num_workers: 0
+  num_workers: 8
 
 rydberg:
   num_states: 2
-  num_encoder_embedding_dims: 3
 
 profiler: 
   profiler: "PyTorchProfiler"
+  advanced_monitoring: False
 
 misc:
   seed: 104
-  prog_bar: True
+  prog_bar: False
diff --git a/config/config_small.yaml b/config/config_small.yaml
@@ -11,16 +11,16 @@ transformer:
 graph_embedding:
   graph_num_layers: 2
   graph_hidden_dim: 64 
-  in_node_dim: 4 # omega delta beta, blockade_radius
+  in_node_dim: 4 # omega delta beta, Rb
 
 training:
   strategy: "auto"
   precision: 32 # half precision saves lots of memory! 32 is default
   max_epochs: 1000
-  batch_size: 32
+  batch_size: 4096
   learning_rate: 0.001
   criterion: "NLLLoss" #KLLoss
-  from_checkpoint: null
+  from_checkpoint: Null
   accumulate_grad_batches: 1
   detect_anomaly: True
 
@@ -34,18 +34,15 @@ optimizer:
   eta_min: 0.00001
 
 dataset:
-  num_atoms: null
-  num_samples: null
-  delta: null
-  num_workers: 0
+  num_workers: 8
 
 rydberg:
   num_states: 2
-  num_encoder_embedding_dims: 3
 
 profiler: 
   profiler: "PyTorchProfiler"
+  advanced_monitoring: False # Only for debugging
 
 misc:
   seed: 104
-  prog_bar: True
+  prog_bar: False