In [w_projector.py](https://github.com/danielroich/PTI/blob/da94d59d15d94822e95840ab5a0aa9ba1a19c851/training/projectors/w_projector.py#L142C1-L142C36) we have a hardcoded size that caused me an error. I encountered this issue when I used 512x512 images. I propose to set it as `w_opt.repeat([1, G.mapping.num_ws, 1])`
In w_projector.py we have a hardcoded size that caused me an error. I encountered this issue when I used 512x512 images.
I propose to set it as
w_opt.repeat([1, G.mapping.num_ws, 1])