neuralmonkey.trainers.test_multitask_trainer module

Unit tests for the multitask 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

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()