This repository contains IMHO Minimal lib for create SAN model, all the code in this lib leverage on this repo.
- Dir lib is actually that minimum lib for create SAN model.
- Dir create_state_dict contains notebook which shows how to convert saved chekpoints in original repo to state_dict that used current minimal lib for loading weights.
You can load already created weights by using notebook in create_state_dict dir.