microsoft / ProphetNet

A research project for natural language generation, containing the official implementations by MSRA NLC team.
MIT License
686 stars 109 forks source link

AR-diffusion: where the code for algorithm 1 is located? #77

Closed miaoGao closed 6 months ago

hwaseem04 commented 6 months ago

Whar do you mean by algorithm 1?

wutong4012 commented 6 months ago

https://github.com/microsoft/ProphetNet/blob/e2c6657309537b94818f5ddbb2a2c5b5559257bf/AR-diffusion/train_utils/resample.py#L124 here :)

miaoGao commented 6 months ago

Whar do you mean by algorithm 1?

I mean "Algorithm 1: Training Process of AR-DIFFUSION" in the paper. image

miaoGao commented 6 months ago

https://github.com/microsoft/ProphetNet/blob/e2c6657309537b94818f5ddbb2a2c5b5559257bf/AR-diffusion/train_utils/resample.py#L124

here :)

Thank you for your reply!