axon
axon copied to clipboard
Support logging in training API
Somewhat related to #21
PyTorch Lightning supports building custom loggers for integration with third-party logging tools (like TensorBoard). We should include a similar API so training can be monitored in tools like TensorBoard.