fix bugs

iMoonLab · Jan 31, 2024 · 6835031 · 6835031
1 parent fede484
commit 6835031
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 2 deletions.
diff --git a/.gitignore b/.gitignore
@@ -1,3 +1,5 @@
+cache/
+
 # Byte-compiled / optimized / DLL files
 __pycache__/
 *.py[cod]

diff --git a/prod_multi_exp.py b/prod_multi_exp.py
@@ -9,7 +9,7 @@
 from utils import MultiExpMetric
 
 
-@hydra.main(config_path=".", config_name="ind_config", version_base="1.1")
+@hydra.main(config_path=".", config_name="prod_config", version_base="1.1")
 def main(cfg: DictConfig):
     logging.info(OmegaConf.to_yaml(cfg))
     res_all = MultiExpMetric()

diff --git a/trans_config.yaml b/trans_config.yaml
@@ -10,7 +10,7 @@ data:
 
 model: 
   # gcn, hgnn, hgnnp, hnhn, unigcn
-  teacher: hgnnp 
+  teacher: hgnn
   # light_hgnn, light_hgnnp
   student: light_hgnnp
   hid: 128