pip install .

curl -fsS -o /tmp/get-poetry.py https://raw.githubusercontent.com/sdispater/poetry/master/get-poetry.py
python3 /tmp/get-poetry.py -y --no-modify-path
export PATH=$HOME/.poetry/bin:$PATH
poetry shell
poetry install

python main.py trainer.epochs=100 data.dataset=tree

python main.py trainer.epochs=100 data.dataset=tree log=true wandb.project=pub_diff wandb.name=tree