align/train.py
src/weakly_sup.py

align/train.py
src/fully_unsup.py

align/models.py
src/models.py