How can I just train and test in Y channel?

[YH-WONG]

I try to modify the VDSR.yaml and DRRN.yaml, set the train_Y: True, but it seems I cannot do that, it will raise the error：

IndexError: too many indices for array: array is 2-dimensional, but 3 were indexed

What should I do?

[seanko29]

@YH-WONG Hi, did you fix it? I seem to get the same error as well.