Question about training dpsr

[1144181135]

Thank you Kai Zhang. I have a question about training dpsr https://arxiv.org/pdf/2008.13751.pdf. About the optimization of eq. 6a and 6b, in the Algorithm 1, it seems that K iterations are in each training sample. However, I can not find the code for K iterations in neither models/network_dpsr.py or models/model_plain.py, Can you point out the part in which the optimization of eq. 6a and 6b ( K iterations in Algorithm 1 )? Thank you !