@ptrblck That’s great. Does it mean in your model the size of output will be same as input size? I can see you used Adam as an optimizer and in the paper which I sent earlier, they used SGD. Any specific reason that you chose this as an optimizer?