Hi,
Pytorch, contrary to Keras/ignite does not provide a builtin training loop. You will have to write it yourself (usually <10 lines) and so you will be able to call all the functions you want in there