DOE OSTI · code-125432
Vistransformers Explained
Abstract
The Vistransformers Explained library is a collection of python notebooks that demonstrate the internal mechanics and uses of visual-transformer (ViT) machine learning models. The code implements, with mild modifications, ViT models that have been made publicly available through publication and GitHub code. The value added by this code is in-depth explanations of the mathematics behind the sub-modules of the ViT models, including original figures. Additionally, the library contains the code necessary to implement and train the ViT models. The library does not include example training data for the models; instead, it would rely on users generating their own datasets. The code is based on the PyTorch python library. It does not include any files other than python scripts, modules, or notebooks.
Keep this discovery
Explore connections, maps & timelines
Callis, Skylar. 2024-01-24. Vistransformers Explained. https://doi.org/10.11578/dc.20240322.8
Cite the original work for its findings. Save a collection to share your selection of sources.