questions about step3 training on single image

[ianyhx]

@MathiasGruber Thank you very much for your excellent code! I have come up against some problems when running your code in Step 3, about training UNet on single image. I've downloaded your pretrained VGG weights and put it in model. However, I found the loss decrease very slow, and the predicted image seems not right.

Epoch 1/10
2000/2000 [==============================] - 1239s 619ms/step - loss: 5.1134 - PSNR: 11.9789

Epoch 2/10
2000/2000 [==============================] - 1226s 613ms/step - loss: 2.8002 - PSNR: 16.5682

Epoch 3/10
2000/2000 [==============================] - 1231s 616ms/step - loss: 2.2371 - PSNR: 19.1468

Is there some wrong with my implementation?

[SunnyPann]

Are you training on your own training set?Where can I download the training model?thank you

[zjlinkin]

I have had the same situation, have you find out why or solved this problem?