update config and remove the randomforest judging

2024-08-13 16:49:53 +02:00
parent b47e83330e
commit d800679b88
3 changed files with 186 additions and 2 deletions
--- a/configs/config.yaml
+++ b/configs/config.yaml
@@ -2,7 +2,7 @@ general:
    name: 'graph_dit'
    wandb: 'disabled' 
    gpus: 1
-    gpu_number: 3
+    gpu_number: 0
    resume: null
    test_only: null
    sample_every_val: 2500
@@ -31,7 +31,7 @@ model:
    lambda_train: [1, 10]  # node and edge training weight 
    ensure_connected: True
 train:
-    n_epochs: 5000
+    n_epochs: 500
    batch_size: 1200
    lr: 0.0002
    clip_grad: null