update the gpu id

configs/config.yaml

@@ -2,6 +2,7 @@ general:
     name: 'graph_dit'
     wandb: 'disabled' 
     gpus: 1
+    gpu_number: 3
     resume: null
     test_only: null
     sample_every_val: 2500
@@ -10,7 +11,7 @@ general:
     chains_to_save: 1
     log_every_steps: 50
     number_chain_steps: 8
-    final_model_samples_to_generate: 10000
+    final_model_samples_to_generate: 100
     final_model_samples_to_save: 20
     final_model_chains_to_save: 1
     enable_progress_bar: False
@@ -30,7 +31,7 @@ model:
     lambda_train: [1, 10]  # node and edge training weight 
     ensure_connected: True
 train:
-    n_epochs: 10000
+    n_epochs: 5000
     batch_size: 1200
     lr: 0.0002
     clip_grad: null