How can change the color and shape parameter of the input picture?

 I want to change the color and shape parameter of the input picture, but I don't know where the code is. So I ask "new bing", it told me to find a file called "train.py"and change the lambda_color and lambda_shape. But I didn't find it. Can you tell me where to change the paramter?