:modulename:`trainers.ForceMatching`
====================================

.. currentmodule:: chemtrain.trainers

.. autoclass:: ForceMatching

   

   
   .. rubric:: Methods

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

   
      ~ForceMatching.__init__
      ~ForceMatching.add_task
      ~ForceMatching.checkpoint
      ~ForceMatching.evaluate
      ~ForceMatching.evaluate_mae_testset
      ~ForceMatching.limit_batches_per_epoch
      ~ForceMatching.load_energy_params
      ~ForceMatching.move_to_device
      ~ForceMatching.predict
      ~ForceMatching.print_training_tasks
      ~ForceMatching.reset_convergence_losses
      ~ForceMatching.restore
      ~ForceMatching.save_energy_params
      ~ForceMatching.save_trainer
      ~ForceMatching.set_batches_per_epoch
      ~ForceMatching.set_dataset
      ~ForceMatching.set_datasets
      ~ForceMatching.set_loader
      ~ForceMatching.train
      ~ForceMatching.update_with_samples
   
   

   
   
   .. rubric:: Attributes

   
   .. autoattribute:: ForceMatching.best_inference_params
      :annotation:
   .. autoattribute:: ForceMatching.best_inference_params_replicated
      :annotation:
   .. autoattribute:: ForceMatching.best_params
      :annotation:
   .. autoattribute:: ForceMatching.energy_fn
      :annotation:
   .. autoattribute:: ForceMatching.params
      :annotation: