.. _tasks:

Task
======

`unimol_tools.tasks <https://github.com/deepmodeling/Uni-Mol/tree/main/unimol_tools/unimol_tools/tasks>`_ oversees the tasks related to the model, such as training and prediction.


Trainer
-------

`unimol_tools.tasks.trainer.py <https://github.com/deepmodeling/Uni-Mol/tree/main/unimol_tools/unimol_tools/tasks/trainer.py>`_ contains the :class:`~unimol_tools.unimol_tools.models.tasks.Trainer`, managing the training, validation, and testing phases.

.. automodule:: unimol_tools.tasks.trainer
   :members:

Split
-------

`unimol_tools.tasks.split.py <https://github.com/deepmodeling/Uni-Mol/tree/main/unimol_tools/unimol_tools/tasks/split.py>`_ manages the split methods in the dataset.

.. automodule:: unimol_tools.tasks.split
   :members: