[train_instruct_pix2pix.py]Fix the LR schedulers when num_train_epochs is passed in a distributed training env

[train_instruct_pix2pix.py]Fix the LR schedulers when num_train_epochs is passed in a distributed training env #24394