:modulename:`trainers.base.DataParallelTrainer`
===============================================

.. currentmodule:: chemtrain.trainers.base

.. autoclass:: DataParallelTrainer

   

   
   .. rubric:: Methods

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

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

   
   
   .. rubric:: Attributes

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