Are you able to post or release the pretrained models from the paper? It would be helpful for those of us that don't have big enough GPUs to train 😆