ConnectionError: Error 10054 while writing to socket. An existing connection was forcibly closed by the remote host

The @classmethod trick worked smoothly, thank you!

An off-topic curiosity:

If my default_num_epochs > max_t and considering that my actual training is happening inside engine.fit, does the training get adjusted based on the max_t param?