Skip to content

EspalomaTemplateGenerator cannot properly load espaloma models trained with dropouts and batch normalization  #263

@kntkb

Description

@kntkb

We need to call model.eval() after loading the espaloma model in generators/template_generators.py#L1580 when loading a model trained with dropouts and batch normalization. It will yield inconsistent inference results if model.eval() is not called.

Reference:
https://pytorch.org/tutorials/beginner/saving_loading_models.html

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions