osc.models.models

Model definitions and wrappers to get attention maps.

Functions

forward_with_attns

Context manager to set and remove hooks that collect attention maps.

Classes

Model	Core model.
ModelOutput	Model output.