Problems with training

[xiaoyi-st]

After reading some of the previous questions, I have another doubt about the data extension of the validation set. Does the validation set have to be in bmp format? In line 60.61 of the main_gan_init program, I have changed the extension of the data set, but it still seems that the data cannot be loaded. The display diagram is as follows, and your help is very much needed.

main_gan_init 59-62lins:

• train_hr_img_list = sorted(tl.files.load_file_list(path=config.TRAIN.hr_img_path, regx='..jpg', printable=False)) train_lr_img_list = sorted(tl.files.load_file_list(path=config.TRAIN.lr_img_path, regx='..jpg', printable=False)) valid_hr_img_list = sorted(tl.files.load_file_list(path=config.VALID.hr_img_path, regx='..jpg', printable=False)) valid_lr_img_list = sorted(tl.files.load_file_list(path=config.VALID.lr_img_path, regx='..jpg', printable=False))

ValueError: Dimensions must be equal, but are 3 and 5 for 'mean_squared_error_1/SquaredDifference' (op: 'SquaredDifference') with input shapes: [5,224,224,3], [5].

[yuinnLIU]

Hello, I met the same problem, did you solve it?