I have seen the torch.utils.trainer, a wrapper for model training, but no docs for it.
torch.utils.trainer
This API will be eventually removed (see GH comment)
You may be interested in using this or this instead.
thanks for your reply