Training task completion condition

[graywolf]

Thank you for providing the nice code.
Under what conditions is the training completed? (Epoch? BCELoss threshold?).
I am not familiar with Pytorch-Lightning description ...