pytorch

estimators

• define_model
  • ZoobotTree
  • get_pytorch_encoder()
  • get_pytorch_dirichlet_head()
• efficientnet_custom
  • custom_top_dirichlet()

training

• train_with_pytorch_lightning
  • train_default_zoobot_from_scratch()
• finetune
  • FinetuneableZoobotAbstract
    • FinetuneableZoobotAbstract.configure_optimizers()
  • FinetuneableZoobotClassifier
  • FinetuneableZoobotRegressor
  • FinetuneableZoobotTree
  • LinearHead
    • LinearHead.forward()
  • load_pretrained_zoobot()
  • get_trainer()
  • download_from_name()
• losses
  • calculate_multiquestion_loss()
  • dirichlet_loss()

predictions

• predict_on_catalog
  • predict()