Skip to content

kurchi1205/Transformer-Optimization

BranchesTags

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

28 Commits

README.md

README.md

 
 

config.py

config.py

 
 

dataset.py

dataset.py

 
 

model.py

model.py

 
 

requirements.txt

requirements.txt

 
 

run_train.py

run_train.py

 
 

run_train_lightning.py

run_train_lightning.py

 
 

train.py

train.py

 
 

train_lightning.py

train_lightning.py

 
 

Repository files navigation

Transformer-Optimization

TODO:

  • Mixed precision in lightning (7secs to 3secs)
  • model to onnx
  • inference code
  • adding proper readme

About

No description, website, or topics provided.

Resources

Readme
Activity

Stars

0 stars

Watchers

1 watching

Forks

0 forks
Report repository

Releases

No releases published

Packages

No packages published

Contributors 2

  •  
  •  

Languages