More info: https://github.com/songlab-cal/gpn

Some params:
* repeat_weight: 0.1
* lr: 120k at 1e-3 + 30k cosine decay