osl_dynamics.models#

Implemented models.

Submodules#

Package Contents#

Functions#

load(dirname[, single_gpu])

Load model from dirname.

Attributes#

models

osl_dynamics.models.models[source]#
osl_dynamics.models.load(dirname, single_gpu=True)[source]#

Load model from dirname.

Parameters:
  • dirname (str) – Path to directory where the config.yml and weights are stored.

  • single_gpu (bool, optional) – Should we compile the model on a single GPU?

Returns:

model – Model object.

Return type:

DyNeMo model