spyrit.core.train
=================

.. automodule:: spyrit.core.train
   :show-inheritance:

   
   
   

   
   
   .. rubric:: Functions

   .. autosummary::
      :toctree:
   
      attr_removal
      attr_transformation
      boxplot
      boxplotconsist
      checkpoint
      compare_model
      count_memory
      count_param
      count_trainable_param
      images_norm
      imshow
      load_net
      multiplot
      read_param
      remove_model_attributes
      rename_model_attributes
      save_net
      tb_profiler
      tb_writer_add_image
      tb_writer_add_scalar
      tb_writer_init
      train_model
      train_model_supervised
      visualize_conv_layers
      visualize_model
   
   

   
   
   .. rubric:: Classes

   .. autosummary::
      :toctree:
      :template: spyrit-class-template.rst
   
      Train_par
      Weight_Decay_Loss