Proyecto EL7008

Hola! Este repositorio contiene los archivos utilizados para implementar y entrenar el modelo Vision Transformer mediante Pytorch y usando el dataset Yoga-82.

A continuación se describe la composición de este repositorio:

• La carpeta model contiene los diferentes módulos del modelo ViT transformer.
• La carpeta utils contiene diferentes módulos necesarios para entrenar el modelo en Pytorch.
• El archivo main.ipynb contiene el código usado para obtener los resultados presentados en Informe.pdf
• El archivo Informe.pdf presenta los principales resultados de haber entrenado Vision Transformer con el dataset Yoga-82.