renate.updaters.learner_components package

Submodules

• renate.updaters.learner_components.component module
  • Component
    • Component.loss()
    • Component.on_train_start()
    • Component.on_train_batch_end()
    • Component.on_load_checkpoint()
    • Component.on_save_checkpoint()
• renate.updaters.learner_components.losses module
  • WeightedLossComponent
    • WeightedLossComponent.loss()
  • WeightedCustomLossComponent
  • WeightedMeanSquaredErrorLossComponent
  • WeightedPooledOutputDistillationLossComponent
  • WeightedCLSLossComponent
    • WeightedCLSLossComponent.on_train_batch_end()
    • WeightedCLSLossComponent.on_load_checkpoint()
    • WeightedCLSLossComponent.on_save_checkpoint()
• renate.updaters.learner_components.reinitialization module
  • ReinitializationComponent
    • ReinitializationComponent.on_train_start()
  • ShrinkAndPerturbReinitializationComponent
    • ShrinkAndPerturbReinitializationComponent.on_train_start()