Could you explain the code a bit more? Currently I’m not sure what shapes some values have, so that re-creating the model is a bit tricky.
It seems upsample is an image tensor. I’m not sure, what h should represent as you are concatenating a one-hot encoded tensor with this image.
Could you clarify it?
I’m sure we can reimplement the model, if you could add some comments to the code (in the best case with shape information!).