git clone https://github.com/Narsil/brownian.git
conda create --name brownian
source activate brownian
pip install torchvision scikit-learn matplotlib tqdm tb-nightly numpy
python train.py
tensorboard --logdir=runs
You can also check the output of the distribution of the output of the transformer
in the figures
directory
The directory checkpoints
will contain the models for future loading.