.. Custom template for autosummary classes. Can be removed when .. https://github.com/sphinx-doc/sphinx/issues/7912 is fixed. dgs.models.engine.engine.EngineModule ===================================== .. currentmodule:: dgs.models.engine.engine .. autoclass:: EngineModule .. rubric:: Methods .. automethod:: EngineModule.__init__ .. automethod:: EngineModule.configure_torch_module .. automethod:: EngineModule.evaluate .. automethod:: EngineModule.get_data .. automethod:: EngineModule.get_hparam_dict .. automethod:: EngineModule.get_target .. automethod:: EngineModule.initialize_optimizer .. automethod:: EngineModule.load_model .. automethod:: EngineModule.predict .. automethod:: EngineModule.print_results .. automethod:: EngineModule.run .. automethod:: EngineModule.save_model .. automethod:: EngineModule.set_model_mode .. automethod:: EngineModule.terminate .. automethod:: EngineModule.test .. automethod:: EngineModule.train_model .. automethod:: EngineModule.validate_params .. automethod:: EngineModule.write_results .. rubric:: Attributes .. autosummary:: :template: attribute.rst :toctree: ~EngineModule.curr_epoch ~EngineModule.device ~EngineModule.is_training ~EngineModule.module_name ~EngineModule.module_type ~EngineModule.name ~EngineModule.name_safe ~EngineModule.precision ~EngineModule.train_load_image_crops ~EngineModule.loss ~EngineModule.model ~EngineModule.writer ~EngineModule.test_dl ~EngineModule.val_dl ~EngineModule.train_dl