Download our dataset from here
TransKnot is a Transformer-based model that classifies knots of different types.
This repository contains the training code in Jupyter Notebook to play with:
train.ipynb
Our model architecture schematics:
KnotFormer is a Transformer-based diffusion model that generates accurate knots of different types
-
To train the diffusion model:
python diffusion_train.py -
To train the classifier:
python classifier_train.py -
To generate: use
generate.ipynb
Our model architecture schematics:
