Hi,  thanks for your reply.
the code is here: https://github.com/ypxie/pytorch-NeuCom/blob/master/tasks/Copy/train.py 52
It does not rely on additional dataset, so it should work with a quick git clone.