TrainStepFunction
=================

.. currentmodule:: penzai.toolshed.basic_training

.. autoclass:: TrainStepFunction
  :members:
  :special-members:
  :show-inheritance:


  
  

  

  
  
  
  
  
  
  
  
    
    
    
    
    
    

  
  .. rubric:: Methods
  .. autosummary::
  
    
    ~TrainStepFunction.__init__
    
    
    ~TrainStepFunction.__call__