/TorchDirectional

Distributions on periodic manifolds for PyTorch

Primary LanguagePythonMIT LicenseMIT

TorchDirectional

Some functions for modeling directional distributions for the use in PyTorch (including GPU). The syntax is similar to that of libDirectional. To test TorchDirectional, create an empty folder, e.g. testFolder, and assign WORKSPACE to it, e.g., via WORKSPACE=testFolder. Then, run the test cases via

wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh -O $WORKSPACE/miniconda.sh
bash $WORKSPACE/miniconda.sh -b -p $WORKSPACE/miniconda
. $WORKSPACE/miniconda/bin/activate
conda env create -f environment.yml
python -m tap | tee taptestresults.tap

Author: Florian Pfaff, pfaff@kit.edu