├── .gitignore ├── LICENSE ├── README.md ├── arch.png ├── codebook.py ├── encoders.py ├── expts ├── run-encoder-ablation.sh ├── run-global-ablation.sh ├── run-hyperparam-sweep-small-experiments.sh ├── run-k-ablation.sh ├── run-lappe-ablation.sh └── run-large-base-experiments.sh ├── local_module.py ├── main_node_ddp.py ├── model.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-stanford/relgt/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-stanford/relgt/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-stanford/relgt/HEAD/README.md -------------------------------------------------------------------------------- /arch.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-stanford/relgt/HEAD/arch.png -------------------------------------------------------------------------------- /codebook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-stanford/relgt/HEAD/codebook.py -------------------------------------------------------------------------------- /encoders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-stanford/relgt/HEAD/encoders.py -------------------------------------------------------------------------------- /expts/run-encoder-ablation.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-stanford/relgt/HEAD/expts/run-encoder-ablation.sh -------------------------------------------------------------------------------- /expts/run-global-ablation.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-stanford/relgt/HEAD/expts/run-global-ablation.sh -------------------------------------------------------------------------------- /expts/run-hyperparam-sweep-small-experiments.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-stanford/relgt/HEAD/expts/run-hyperparam-sweep-small-experiments.sh -------------------------------------------------------------------------------- /expts/run-k-ablation.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-stanford/relgt/HEAD/expts/run-k-ablation.sh -------------------------------------------------------------------------------- /expts/run-lappe-ablation.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-stanford/relgt/HEAD/expts/run-lappe-ablation.sh -------------------------------------------------------------------------------- /expts/run-large-base-experiments.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-stanford/relgt/HEAD/expts/run-large-base-experiments.sh -------------------------------------------------------------------------------- /local_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-stanford/relgt/HEAD/local_module.py -------------------------------------------------------------------------------- /main_node_ddp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-stanford/relgt/HEAD/main_node_ddp.py -------------------------------------------------------------------------------- /model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-stanford/relgt/HEAD/model.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snap-stanford/relgt/HEAD/utils.py --------------------------------------------------------------------------------