README for EE541 final project：

enviroment set up

CNNmodel.ipynb :  native env      3900X + rtx2080super


description：our own design of the cnn model. wangnet.py is the python file store the architecture of this CNN model. 


VITmodel.ipynb :  colab pro      Xeon  +  P100 

description：Vision transformer model using transfer learning, requires timm package support.


XGboost_baseline.ipynb: native cpu i9 9800H

description：baseline model using machine learning