Skip to content

Release 0.0.8

Compare
Choose a tag to compare
@clementchadebec clementchadebec released this 07 Sep 17:03
· 41 commits to main since this release
2a782b9

New Features:

  • Added MLFlowCallback in TrainingCalbacks further to #44
  • Allow custom Dataset inheriting from torch.utils.data.Dataset to be passed as inputs in the training_pipeline further to #35
def __call__(
        self,
        train_data: Union[np.ndarray, torch.Tensor, torch.utils.data.Dataset],
        eval_data: Union[np.ndarray, torch.Tensor, torch.utils.data.Dataset] = None,
        callbacks: List[TrainingCallback] = None,
    ):

Minor changes

  • Unify data handling in FactorVAE with other models. (half of the batch is used for reconstruction and the other one for factorial representation)
  • Change model sanity check method in trainers (use loaders in check instead of datasets)
  • Add encoder/decoder losses needed in CoupledOptimizerTrainer and update tests