Hi,
Currently, your code is only valid for solving classification problems. Can you modify/try it with some regression data on grayscale images?
For example input to the network is a grayscale image and output are some numbers like:
car.png 0.1 0.5 0.47 0.64 0.54
Thank you.