dlk.managers package

Submodules

dlk.managers.lightning module

class dlk.managers.lightning.LightningManager(config: dlk.managers.lightning.LightningManagerConfig, rt_config: Dict)[source]

Bases: object

pytorch-lightning traning manager

fit(**inputs)[source]

fit the model and datamodule to trainer

Parameters

**inputs – dict of input, include “model”, ‘datamodule’

Returns

Undefine

get_callbacks(callback_configs: List[Dict], rt_config: Dict)[source]

init the callbacks and return the callbacks list

Parameters

  • callback_configs – the config of every callback

  • rt_config – {“save_dir”: ‘..’, “name”: ‘..’}

Returns

all callbacks

predict(**inputs)[source]

fit the model and datamodule.predict_dataloader to predict

Parameters

**inputs – dict of input, include “model”, ‘datamodule’

Returns

predict list

test(**inputs)[source]

fit the model and datamodule.test_dataloader to test

Parameters

**inputs – dict of input, include “model”, ‘datamodule’

Returns

Undefine

validate(**inputs)[source]

fit the model and datamodule.validation to validate

Parameters

**inputs – dict of input, include “model”, ‘datamodule’

Returns

Undefine

class dlk.managers.lightning.LightningManagerConfig(config)[source]

Bases: dlk.utils.config.BaseConfig

docstring for LightningManagerConfig check https://pytorch-lightning.readthedocs.io trainer for paramaters detail

get_callbacks_config(config: Dict) → List[Dict][source]

get the configs for callbacks

Parameters

config – {“config”: {“callbacks”: [“callback_names”..]}, “callback@callback_names”: {config}}

Returns

configs which name in config[‘config’][‘callbacks’]

Module contents

managers

dlk.managers.import_managers(managers_dir, namespace)[source]