Skip to content
/ Self-Attention Public

simple implements Non-Local Neural Networks for image classification(Fashion-Mnist)

13 stars 6 forks Branches Tags Activity
Star
Notifications You must be signed in to change notification settings

lovekittynine/Self-Attention

BranchesTags

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
README.md
README.md
 
 
SA_train.py
SA_train.py
 
 
basic_model.py
basic_model.py
 
 
basic_train.py
basic_train.py
 
 
dataset.py
dataset.py
 
 
self_attetion_model.py
self_attetion_model.py
 
 

Repository files navigation

Self-Attention

Pytorch

I have test self-attention in FashionMnist classification,and Basic Model Accuracy=0.913, Self-Attention Model=0.912

Just for fun!!!

Maybe there exist some details that I dont watch,if you feel some errors welcome to contact me!!! simple implements Non-Local Neural Networks for image classification(Fashion-Mnist)

About

simple implements Non-Local Neural Networks for image classification(Fashion-Mnist)

Resources

Readme
Activity

Stars

13 stars

Watchers

0 watching

Forks

6 forks
Report repository

Releases

No releases published

Packages

No packages published

Languages