-
encoder.py : torch encoder model
-
decoder.py : torch decoder model
-
train.py : training the model
-
utils.py : some important functions which are used through out
-
data.py : dataloaders and stuff
-
transform.py : Not sure!
-
Test transform.py