Task

unimol_tools.tasks oversees the tasks related to the model, such as training and prediction.

Trainer

unimol_tools.tasks.trainer.py contains the Trainer, managing the training, validation, and testing phases.