API
- Callbacks
- Datasets
- Learners
- ClassificationLearner
- RecognitionLearner
- SegmentationLearner
- DetectionLearner
- GANLearner
BaseLearner
BaseLearner.model
BaseLearner.loss
BaseLearner.optimizer
BaseLearner.lr_scheduler
BaseLearner.train_metrics
BaseLearner.val_metrics
BaseLearner.test_metrics
BaseLearner.data_keys
BaseLearner.target_keys
BaseLearner.get_outputs()
BaseLearner.batch
BaseLearner.get_targets()
BaseLearner.batch
BaseLearner.loss_step()
BaseLearner.outputs
BaseLearner.targets
BaseLearner.on_test_epoch_end()
BaseLearner.on_train_epoch_end()
BaseLearner.on_validation_epoch_end()
- Losses
- Lr schedulers
- Metrics
- Optimizers