The paper mentions the results on agd20k. I would like to ask how to reproduce the results of the paper on the agd20k dataset.