The learning rate scheduler
CosineAnnealingWarmRestarts does not accept a
T_mult as a float number.
Is there a reason why? Is there a workaruond? It would be very useful to have it as a float to have more control over the rate at which the period changes.