Update vit.py to include LayerNorm in the MLP head which is missing#317
Open
vivekh2000 wants to merge 1 commit intolucidrains:mainfrom
Open
Update vit.py to include LayerNorm in the MLP head which is missing#317vivekh2000 wants to merge 1 commit intolucidrains:mainfrom
vivekh2000 wants to merge 1 commit intolucidrains:mainfrom
