search

victorca25 / traiNNer

traiNNer: Deep learning framework for image and video super-resolution, restoration and image-to-image translation, for training and testing.
Apache License 2.0
293 stars 39 forks source link

Pixel Unshuffle is broken #53

Closed Kim2091 closed 1 year ago

Kim2091 commented 3 years ago

Training a 1x model with Pixel Unshuffle (using the supplied pretrained model) yields this error:

[Python] RuntimeError: Given groups=1, weight of size [64, 48, 3, 3], expected input[1, 4, 297, 397] to have 48 channels, but got 4 channels instead [ESRGAN] Upscaling Error: Index was outside the bounds of the array. at Cupscale.PreviewMerger.Merge() at Cupscale.Main.Upscale.<Run>d__8.MoveNext()

victorca25 commented 3 years ago

This is an error in Cupscale, is it the correct error?

Kim2091 commented 3 years ago

The error occurred in all forks of ESRGAN when trying to inference with the model. It was specifically about the weights, I didn't mean to copy the Cupscale line