Performs a single optimization step. Arguments: closure (callable, optional): A closure that reevaluates the model

Step could be called after every batch update Example: >>> scheduler = CosineAnnealingWarmRestarts(optimizer, T_0, T_mult)

Performs a single optimization step across all parameter groups of the model. This method iterates over each parameter group, updatin

Performs a single optimization step. Arguments: gradsH: The gradient used to compute Hessian vector product. closure (

Performs a single optimization step. Arguments: closure (callable, optional): A closure that reevaluates the model

Performs a single optimization step. Arguments: closure (callable, optional): A closure that reevaluates the model

Performs a single optimization step. Arguments: closure (callable, optional): A closure that reevaluates the model

r"""Wide ResNet-101-2 model from `"Wide Residual Networks" <https://arxiv.org/pdf/1605.07146.pdf>`_ The model is the same as ResNet except fo

r"""Wide ResNet-50-2 model from `"Wide Residual Networks" <https://arxiv.org/pdf/1605.07146.pdf>`_ The model is the same as ResNet except for