Hooks
Inference
Learning
Hooks can be used to modify the learning process
- XPM Configxpmir.learning.context.TrainingHook[source]
Bases:
Config
Base class for all training hooks
- XPM Configxpmir.learning.context.InitializationTrainingHook[source]
Bases:
InitializationHook
Base class for hooks called at each epoch (before/after)
Distributed
Hooks can be used to distribute a model over GPUs
- XPM Configxpmir.distributed.DistributableModel[source]
Bases:
Config
A model that can be distributed over GPUs
Subclasses must implement
distribute_models()
- XPM Configxpmir.distributed.DistributedHook(*, models)[source]
Bases:
InitializationHook
Hook to distribute the model processing
When in multiprocessing/multidevice, use torch.nn.parallel.DistributedDataParallel ,otherwise use torch.nn.DataParallel.
- models: List[xpmir.distributed.DistributableModel]
The model to distribute over GPUs