I’m sorry but it is not the case.
Note that dataset is dynamically generated by part A during training. It is not fixed.
In my description, dataset changes its data every 10 epochs. However, in your code, dataset is unchanged.
Thanks for your reply!