Callback in pytorch

Hello everyone. How can I use callback in pytorch for saving end of epoch in google colab?

Higher-level APIs such as Ignite or Lightning provide these type of hooks, so you should check them out. :wink: