dicee.trainer.model_parallelism
===============================

.. py:module:: dicee.trainer.model_parallelism


Classes
-------

.. autoapisummary::

   dicee.trainer.model_parallelism.TensorParallel


Functions
---------

.. autoapisummary::

   dicee.trainer.model_parallelism.extract_input_outputs
   dicee.trainer.model_parallelism.find_good_batch_size
   dicee.trainer.model_parallelism.forward_backward_update_loss


Module Contents
---------------

.. py:function:: extract_input_outputs(z: list, device=None)

.. py:function:: find_good_batch_size(train_loader, tp_ensemble_model)

.. py:function:: forward_backward_update_loss(z: Tuple, ensemble_model) -> float

.. py:class:: TensorParallel(args, callbacks)

   Bases: :py:obj:`dicee.abstracts.AbstractTrainer`


   Abstract class for Trainer class for knowledge graph embedding models


   Parameter
   ---------
   args : str
       ?

   callbacks: list
           ?


   .. py:method:: fit(*args, **kwargs)

      Train model