neuralmonkey.trainers.test_multitask_trainer module

Unit tests for the multitask trainer.

class neuralmonkey.trainers.test_multitask_trainer.DummyObjective(name: str, decoder: neuralmonkey.trainers.test_multitask_trainer.TestMP) → None

Bases: neuralmonkey.trainers.objective.Objective

__init__(name: str, decoder: neuralmonkey.trainers.test_multitask_trainer.TestMP) → None

Construct the objective.

Parameters:name – The name for the objective. This will be used e.g. in TensorBoard.

loss

Return the loss tensor fetched by the trainer.

class neuralmonkey.trainers.test_multitask_trainer.TestMP(name: str, reuse: Union[neuralmonkey.model.model_part.ModelPart, NoneType] = None, save_checkpoint: str = None, load_checkpoint: str = None, initializers: List[Tuple[str, Callable]] = None) → None

Bases: neuralmonkey.model.model_part.ModelPart

loss

var

class neuralmonkey.trainers.test_multitask_trainer.TestMultitaskTrainer(methodName='runTest')

Bases: unittest.case.TestCase

setUp()

Hook method for setting up the test fixture before exercising it.

classmethod setUpClass()

Hook method for setting up class fixture before running tests in the class.

test_mt_trainer()