How to use your callbacks.py

[riven314]

Hi! I am looking for a way to implement callbacks in PyTorch, so as to mimic the way I write training pipeline in Keras.
I find your repo very comprehensive and pretty related to the problem (particularly for torchsample/callbacks.py) that I would like to take reference from that.
Would you mind giving me an example of how you use your callbacks module in your training loop? Thank you!

Alex