You need poetry.
- Create a new envronment
- Install PyTorch - this depends on your machine, so you need to choose the right one for you.
poetry install
poetry build
poetry install
(again)
The last 2 steps are for adding a shortcut so that you can run the model with run
run
run data.batch_size=128
run data/schema=adult data=adu