Why does the color change？

[watertianyi]

[RisingEntropy]

I also have encountered this, and after I train more epoches, the whole image goes into totally black.

[RisingEntropy]

I have solved my own problem. In my case, the change in the color is caused by under-fitting. And the black proble is caused by a wrong cell value. Hope my experience can solve you problem.

[home56200]

How is the black problem solved? Does under-fitting mean training for a long time, or changing certain parameters? Thank you.

[RisingEntropy]

I just trained more epoches and the problem was solved. (I was just verifying my code, so I just trained 5 epoches at first)

[xiximelon]

@RisingEntropy Hello! May I know how long does the code takes to train 1 epoch? Thanks in advance!