serotiny.models package

Subpackages

• serotiny.models.utils package
  • Submodules
    • serotiny.models.utils.mlflow module
      • get_config()
      • load_model_from_checkpoint()

Submodules

• serotiny.models.base_model module
  • BaseModel
    • BaseModel.configure_optimizers()
    • BaseModel.forward()
    • BaseModel.parse_batch()
    • BaseModel.predict_step()
    • BaseModel.test_epoch_end()
    • BaseModel.test_step()
    • BaseModel.train_epoch_end()
    • BaseModel.training_step()
    • BaseModel.validation_epoch_end()
    • BaseModel.validation_step()
• serotiny.models.basic_model module
  • BasicModel
    • BasicModel.forward()
    • BasicModel.parse_batch()