def configure_optimizers(self):
    params = ((list(self.backbone.parameters()) + list(self.projection_head.parameters())) + list(self.prediction_head.parameters()))
    optim = torch.optim.SGD(params, lr=(0.06 * lr_factor), momentum=0.9, weight_decay=0.0005)
    scheduler = torch.optim.lr_scheduler.CosineAnnealingLR(optim, max_epochs)
    return ([optim], [scheduler])
def configure_optimizers(self):
    params = ((list(self.backbone.parameters()) + list(self.projection_head.parameters())) + list(self.prediction_head.parameters()))
    optim = torch.optim.SGD(params, lr=(0.06 * lr_factor), momentum=0.9, weight_decay=0.0005)
    scheduler = torch.optim.lr_scheduler.CosineAnnealingLR(optim, max_epochs)
    return ([optim], [scheduler])