wbhu/DnCNN-tensorflow

The training time

wwhappylife opened this issue · 1 comments

wwhappylife commented

I use gtx1080 to train, and didn't change your code, but it trained too slow, almost three hours to train a epoch

wbhu commented

Hi @wwhappylife ,

Actually, I set stride to be 35 in generate_patches.py, and trained only 8k iterations as README shows. It trained very fast. I have post my trained model on the repo.

Thanks

  • 👍1