modlee.model package

Submodules

• modlee.model.callbacks module
  • DataMetafeaturesCallback
    • DataMetafeaturesCallback.on_train_start()
  • LogCodeTextCallback
    • LogCodeTextCallback.on_train_start()
    • LogCodeTextCallback.setup()
  • LogONNXCallback
    • LogONNXCallback.on_train_start()
    • LogONNXCallback.setup()
  • LogOutputCallback
  • LogParamsCallback
    • LogParamsCallback.on_train_start()
  • LogTransformsCallback
    • LogTransformsCallback.on_train_start()
  • ModleeCallback
    • ModleeCallback.get_input()
  • PushServerCallback
    • PushServerCallback.on_fit_end()
• modlee.model.image_model module
  • ImageCallback
    • ImageCallback.on_validation_batch_end()
  • ModleeImageModel
    • ModleeImageModel.configure_callbacks()
• modlee.model.model module
  • ModleeModel
    • ModleeModel.configure_callbacks()
    • ModleeModel.run_path
  • SimpleDataset
  • SimpleModel
    • SimpleModel.configure_optimizers()
    • SimpleModel.forward()
    • SimpleModel.training_step()
• modlee.model.recommended_model module
  • RecommendedModel
    • RecommendedModel.configure_callbacks()
    • RecommendedModel.configure_optimizers()
    • RecommendedModel.forward()
    • RecommendedModel.on_train_epoch_end()
    • RecommendedModel.training_step()
    • RecommendedModel.validation_step()

Module contents