Update https://github.com/Akettle44/SRGAN-PyTorch/blob/main/src/model/model.py#L26 to match local fix from a while ago