-
Notifications
You must be signed in to change notification settings - Fork 447
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Training for pytorch #60
Comments
The PyTorch version is an approximate reimplementation and performs a bit differently. I will have some more time after CVPR deadline to look into it further. Thanks for your patience. |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Hi All,
First of all thanks for the amazing project. I got it more or less working (see my how to in #10).
Thing is the caffe results seem much better than the pytorch one. Better in the sense that caffe colors in the bands of the mortar which pytorch colors outside the bands as well). But the caffe version runs out of memory with a predicted requiirement of 10GB (6.7GB used in cpu_mode) while the pytorch version uses only 1GB. I have followed the data training steps from colorization-pytorch (downloaded pretrained model) but this doesn't seem to change much.
Anything I am missing?
The descriptions on the main site:
and
didn't really help much. Any clarification would be much appreciated.
Yours,
Frank
The text was updated successfully, but these errors were encountered: