siml.update_functions package¶
Submodules¶
siml.update_functions.element_batch_update module¶
- class siml.update_functions.element_batch_update.ElementBatchUpdate(element_batch_size: int, loss_func: ILossCalculator, other_loss_func: Callable, split_data_func: Callable, clip_grad_value: float | None = None, clip_grad_norm: float | None = None)¶
Bases:
IStepUpdateFunction
siml.update_functions.pseudo_batch_update module¶
- class siml.update_functions.pseudo_batch_update.Counter(base_value: int)¶
Bases:
object- increment() None¶
- property is_full: bool¶
- property value: int¶
- class siml.update_functions.pseudo_batch_update.PseudoBatchStep(batch_size: int, loss_func: ILossCalculator, other_loss_func: Callable, split_data_func: Callable, device: str, output_device: str, loss_slice: slice, time_series_split: bool, clip_grad_value: float | None = None, clip_grad_norm: float | None = None)¶
Bases:
IStepUpdateFunction
siml.update_functions.standard_update module¶
- class siml.update_functions.standard_update.StandardUpdate(loss_func: ILossCalculator, other_loss_func: Callable, split_data_func: Callable, device: str, output_device: str, loss_slice: slice, time_series_split: bool, clip_grad_value: float | None = None, clip_grad_norm: float | None = None)¶
Bases:
IStepUpdateFunction
siml.update_functions.update_interface module¶
- class siml.update_functions.update_interface.IStepUpdateFunction¶
Bases:
object