:modulename:`trainers.Difftre`
==============================

.. currentmodule:: chemtrain.trainers

.. autoclass:: Difftre

   

   
   .. rubric:: Methods

   .. autosummary::
      :toctree: _autosummary
      :template: method.rst

   
      ~Difftre.__init__
      ~Difftre.add_statepoint
      ~Difftre.add_task
      ~Difftre.checkpoint
      ~Difftre.get_sim_state
      ~Difftre.init_step_size_adaption
      ~Difftre.load_energy_params
      ~Difftre.move_to_device
      ~Difftre.predict
      ~Difftre.print_training_tasks
      ~Difftre.restore
      ~Difftre.save_energy_params
      ~Difftre.save_trainer
      ~Difftre.train
   
   

   
   
   .. rubric:: Attributes

   
   .. autoattribute:: Difftre.best_params
      :annotation:
   .. autoattribute:: Difftre.energy_fn
      :annotation:
   .. autoattribute:: Difftre.params
      :annotation: