Default size is 256x256 if the input size is 512x512. But how can I make it be the same size as input? Thank you.