How can I just train and test in Y channel? #2
Description
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?