pyabsa.tasks._Archive.ProteinRegression.trainer.proteinr_trainer
Module Contents
Classes
Trainer class for training PyABSA models |
- class pyabsa.tasks._Archive.ProteinRegression.trainer.proteinr_trainer.ProteinRTrainer(config: pyabsa.tasks._Archive.ProteinRegression.configuration.proteinr_configuration.ProteinRConfigManager = None, dataset=None, from_checkpoint: str = None, checkpoint_save_mode: int = ModelSaveOption.SAVE_MODEL_STATE_DICT, auto_device: Union[bool, str] = DeviceTypeOption.AUTO, path_to_save=None, load_aug=False)[source]
Bases:
pyabsa.framework.trainer_class.trainer_template.Trainer
Trainer class for training PyABSA models