Deprecated paramater

[GiuseppeLeviBo]

Hi,
I'm testing your SinGAN implementation with PyTorch 2.01 (using the pytorch/pytorch:latest docker image) and seems working nicely.
The is only a Warning

/opt/conda/lib/python3.10/site-packages/torch/optim/lr_scheduler.py:152: UserWarning: The epoch parameter in `scheduler.step()` was not necessary and is being deprecated where possible. Please use `scheduler.step()` to step the scheduler. During the deprecation, if epoch is different from None, the closed form is used instead of the new chainable form, where available. Please open an issue if you are unable to replicate your use case: https://github.com/pytorch/pytorch/issues/new/choose.
  warnings.warn(EPOCH_DEPRECATION_WARNING, UserWarning)

is not clear to me if it refers to your code or something internal to Torch itself but I think that you would be interested knowing.