vipr_reflectometry.flow_models.load_model package

Subpackages

• vipr_reflectometry.flow_models.load_model.models package
  • Subpackages
    • vipr_reflectometry.flow_models.load_model.models.networks package
      • Submodules
      • vipr_reflectometry.flow_models.load_model.models.networks.flow_networks module
      • Module contents
  • Module contents

Submodules

vipr_reflectometry.flow_models.load_model.flow_model_loader module

class vipr_reflectometry.flow_models.load_model.flow_model_loader.FlowModelLoader(**kw: Any)

Bases: ModelLoaderHandler

Model loader for HZDR Flow Network models.

Supports all flow-based architectures: - CINNWithConvEmb - NSFWithConvEmb - MAFNetworkWithConvEmb - And other flow models from flow_networks.py

class Meta

Bases: object

label = 'flow_model_loader'

app: VIPR

class vipr_reflectometry.flow_models.load_model.flow_model_loader.FlowModelLoaderParams(*, config_name: str, config_url: str | None = None, weights_url: str | None = None, model_dir: str = 'saved_models', device: str | None = None, weights_format: str = 'pt')

Bases: BaseModel

Parameters for the flow model loader.

config_name: str

config_url: str | None

device: str | None

model_config: ClassVar[ConfigDict] = {'extra': 'forbid'}

Configuration for the model, should be a dictionary conforming to [ConfigDict][pydantic.config.ConfigDict].

model_dir: str

weights_format: str

weights_url: str | None

Module contents