cross-entropy loss over the retain-set missing in the code.

[kyunyoo]

the paper mentions that you guys used cross-entropy loss over the retain-set, but I could was not able to find them in the code, so I was wondering the result in the paper was made with cross-entropy loss over the retain-set or not.