hsun29 / deeplearning

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

deeplearning/CS584 Lab Illinois Institute Tech

deeplearning ViT & Linformer mixed model

1, You must download all the folder to your local env, the folder include linformer model, the main code of mixed model and dataset.

2, You may do not need to modify the structre of the alz_data folder. You must modify the path in vit.py after you modified the structre.

3, The model achieve around 96% in val data.

4, This branch do not include plot function, but there generate training history after you finish training, you can analysis it.

5, This branch will continue to be completed...

About


Languages

Language:Python 100.0%