update some score function

2024-08-05 21:45:15 +02:00
parent f5d00be56e
commit 205f43291b
4 changed files with 37 additions and 17 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: 2
    resume: null
    test_only: null
    sample_every_val: 2500
@@ -31,7 +31,8 @@ model:
    lambda_train: [1, 10]  # node and edge training weight 
    ensure_connected: True
 train:
-    n_epochs: 5000
+    # n_epochs: 5000
+    n_epochs: 10
    batch_size: 1200
    lr: 0.0002
    clip_grad: null