ride.lifecycle
¶
Module Contents¶
Classes¶
Adds train, val, and test lifecycle methods with cross_entropy loss |
Functions¶
|
|
|
Attributes¶
- class ride.lifecycle.Lifecycle(hparams=None, *args, **kwargs)[source]¶
Bases:
ride.metrics.MetricMixin
Adds train, val, and test lifecycle methods with cross_entropy loss
During its traning_epoch_end(epoch) lifecycle method, it will call on_traning_epoch_end for all superclasses of its child class
- forward: Callable[[torch.Tensor], torch.Tensor][source]¶
- static configs() ride.core.Configs [source]¶
- metrics_step(preds: torch.Tensor, targets: torch.Tensor, **kwargs) ride.metrics.MetricDict [source]¶
- ride.lifecycle.detach_to_cpu(x: Union[torch.Tensor, Sequence[torch.Tensor], Dict[Any, torch.Tensor]])[source]¶
- ride.lifecycle.cat_steps(steps: Sequence[Union[torch.Tensor, Sequence[torch.Tensor], Dict[Any, torch.Tensor]]])[source]¶