Paper version

This version allows to use the weights of the checkpoints provided. If you want to reproduce the results of the paper, please use this release.