pyabsa.tasks._Archive.RNAClassification.trainer.rnac_trainer

Module Contents

Classes

RNACTrainer

Trainer class for training PyABSA models

class pyabsa.tasks._Archive.RNAClassification.trainer.rnac_trainer.RNACTrainer(config: pyabsa.tasks._Archive.RNAClassification.configuration.rnac_configuration.RNACConfigManager = None, dataset=None, from_checkpoint: str = None, checkpoint_save_mode: int = ModelSaveOption.SAVE_MODEL_STATE_DICT, auto_device: bool | str = DeviceTypeOption.AUTO, path_to_save=None, load_aug=False)

Bases: pyabsa.framework.trainer_class.trainer_template.Trainer

Trainer class for training PyABSA models