clarification of parameters

Hi all,

Thanks for this great project. I need clarification on a few parameters:

my training images are of size 512x512, what should I set:

train_size,
input_height,
input_width,
output_height,
output_width

I had input_height = 512, since I was thinking that's the height/width of my input images. But I also saw that train_size is an optional parameter. It'd be great if someone can clarify these parameters for me :) Thanks in advance and have a great day.