How to replicate the results in the paper?

[yyoshikai]

Hello, thank you for providing the codes.

I was trying to replicate the results in the paper about generative performance on MOSES dataset, but I failed. Which weight in https://www.kaggle.com/datasets/virajbagal/ligflow-final-weights should be used? I tried moses_newtokens.pt and moses_scaf_wholeseq_newtokens.pt and did

python generate/generate.py --model_weight moses_newtokens.pt --csv_name generate --gen_size 30000

python generate/generate.py --model_weight moses_scaf_wholeseq_newtokens.pt --csv_name moses_scaf_wholeseq_newtokens --gen_size 30000 --scaffold

but FCD/Test and FCD/TestSF did not reach the results in Table 1. Am I forgetting some arguments?

Thank you very much!