diff --git a/.gitignore b/.gitignore index 68bc17f..132c383 100644 --- 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 index bfcb331..3788184 100644 --- 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 index d64633a..7f42c55 100644 --- 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