hy2dl.training

hy2dl.training.get_loss(cfg: Config)

Get loss object, depending on the run configuration.

Parameters:

cfg (Config) – The run configuration.

Returns:

A new loss instance of the type specified in the config.

Return type:

loss_module.BaseLoss

hy2dl.training.get_loss_registry()

• BaseTrainer
  • BaseTrainer
    • BaseTrainer.train_model()
• Loss functions
  • BaseLoss
    • BaseLoss.forward()
  • NLL
    • NLL.forward()
  • NSEBasinAveraged
    • NSEBasinAveraged.forward()
  • WeightedMSE
    • WeightedMSE.forward()
• Optimizer
  • Optimizer
    • Optimizer.clip_grad_and_step()
    • Optimizer.update_optimizer_lr()