The model is trained on popular INCLUDE Dataset created by AI4Bharat, IIT Madras. The architecture is a combination of EfficientNet + Bert Encoder layer which contains self attention mechanism. The model is able to obtain a Test Accuracy of 91.17%. When using VGG16 + Bert Encoder layer, we obtain an accuracy of 92.7%. The git repository provides implementation with EfficientNet + Bert Encoder.
-
Notifications
You must be signed in to change notification settings - Fork 0
saproovarun/ISLTransformer
Folders and files
Name | Name | Last commit message | Last commit date | |
---|---|---|---|---|
Repository files navigation
About
No description, website, or topics provided.
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published