TorchTensorBoard
================

.. currentmodule:: lib.training.tensorboard

.. autoclass:: TorchTensorBoard
   :show-inheritance:

   .. rubric:: Attributes Summary

   .. autosummary::

      ~TorchTensorBoard.model

   .. rubric:: Methods Summary

   .. autosummary::

      ~TorchTensorBoard.on_batch_begin
      ~TorchTensorBoard.on_batch_end
      ~TorchTensorBoard.on_epoch_begin
      ~TorchTensorBoard.on_epoch_end
      ~TorchTensorBoard.on_predict_batch_begin
      ~TorchTensorBoard.on_predict_batch_end
      ~TorchTensorBoard.on_predict_begin
      ~TorchTensorBoard.on_predict_end
      ~TorchTensorBoard.on_save
      ~TorchTensorBoard.on_test_batch_begin
      ~TorchTensorBoard.on_test_batch_end
      ~TorchTensorBoard.on_test_begin
      ~TorchTensorBoard.on_test_end
      ~TorchTensorBoard.on_train_batch_begin
      ~TorchTensorBoard.on_train_batch_end
      ~TorchTensorBoard.on_train_begin
      ~TorchTensorBoard.on_train_end
      ~TorchTensorBoard.set_model
      ~TorchTensorBoard.set_params

   .. rubric:: Attributes Documentation

   .. autoattribute:: model

   .. rubric:: Methods Documentation

   .. automethod:: on_batch_begin
   .. automethod:: on_batch_end
   .. automethod:: on_epoch_begin
   .. automethod:: on_epoch_end
   .. automethod:: on_predict_batch_begin
   .. automethod:: on_predict_batch_end
   .. automethod:: on_predict_begin
   .. automethod:: on_predict_end
   .. automethod:: on_save
   .. automethod:: on_test_batch_begin
   .. automethod:: on_test_batch_end
   .. automethod:: on_test_begin
   .. automethod:: on_test_end
   .. automethod:: on_train_batch_begin
   .. automethod:: on_train_batch_end
   .. automethod:: on_train_begin
   .. automethod:: on_train_end
   .. automethod:: set_model
   .. automethod:: set_params