urbanlc.model.deep_learning.OLITIRSDeepLearning.setup_trainer

OLITIRSDeepLearning.setup_trainer(loss_fn_params: Dict[str, Any], optimizer_params: Dict[str, Any], scheduler_params: Dict[str, Any], **kwargs: Any) → None

Set up the trainer with loss functions, optimizer, and scheduler.

Parameters:

• loss_fn_params (Dict[str, Any]) – Dictionary containing loss function parameters.

• optimizer_params (Dict[str, Any]) – Dictionary containing optimizer parameters.

• scheduler_params (Dict[str, Any]) – Dictionary containing scheduler parameters.