Comments (6)
Issue now resolved, please see the latest code pushed to the repository today. I have also uploaded a model from epoch 99 that is more accurate (better PSNR, SSIM).
from curl.
Hi, CPU mode is straightforward. You just need to place the cuda keywords in the code with cpu I believe e.g. net.cuda() with net.cpu() etc. For inference you'll also need to load the model on the CPU, more information is here: https://pytorch.org/tutorials/beginner/saving_loading_models.html. Feel free to raise a PR if you have a CPU/GPU switch implemented for CURL.
from curl.
Hi, thank you for your interest in our work. I will have updated code in a few days time, including a new pre-trained model.
from curl.
Yes, I am using the updated commit. Here is the command I use which gives the error
python3.8 main.py --inference_img_dirpath=./adobe5k_dpe/ --checkpoint_filepath=./pretrained_models/curl_validpsnr_23.070664790618277_validloss_0.029170328751206398_testpsnr_23.556465119481636_testloss_0.02679653838276863_epoch_24_model.pt
from curl.
Thank you very much for the fix! By the way, does your code support only GPU, or I can also run on CPU mode?
from curl.
Got it, thank you!
from curl.
Related Issues (20)
- how to train the raw data(unit16) to jpg without quality loss HOT 1
- FileNotFoundError: [Errno 2] No such file or directory: './pretrained_models/curl_validpsnr_23.073045286204017_validloss_0.0701291635632515_testpsnr_23.584083321292365_testloss_0.061363041400909424_epoch_510_model.pt' HOT 2
- it seems the batch size must be 1? HOT 1
- Problems when using images of size 256x256 HOT 2
- Lightroom image exporting HOT 2
- hi,is it normal for loss HOT 1
- RuntimeError: Given groups=1, weight of size [16, 3, 3, 3], expected input[1, 4, 343, 514] to have 3 channels, but got 4 channels instead HOT 1
- ZeroDivisionError: float division by zero HOT 2
- The weights and the architecture are not the same HOT 1
- running code without CUDA HOT 19
- Batch size > 1 version HOT 2
- Inference Pretrained Model with CPP scratch HOT 1
- inference of test images HOT 1
- Bigger Batch size HOT 1
- Can't reproduce results HOT 1
- Training/Validation/Test splits link is down HOT 1
- Doubt about results HOT 4
- 请教一下这块实现的原理 HOT 2
- dim HOT 1
- Pretrained model results
Recommend Projects
-
React
A declarative, efficient, and flexible JavaScript library for building user interfaces.
-
Vue.js
🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
-
Typescript
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
-
TensorFlow
An Open Source Machine Learning Framework for Everyone
-
Django
The Web framework for perfectionists with deadlines.
-
Laravel
A PHP framework for web artisans
-
D3
Bring data to life with SVG, Canvas and HTML. 📊📈🎉
-
Recommend Topics
-
javascript
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
-
web
Some thing interesting about web. New door for the world.
-
server
A server is a program made to process requests and deliver data to clients.
-
Machine learning
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
-
Visualization
Some thing interesting about visualization, use data art
-
Game
Some thing interesting about game, make everyone happy.
Recommend Org
-
Facebook
We are working to build community through open source technology. NB: members must have two-factor auth.
-
Microsoft
Open source projects and samples from Microsoft.
-
Google
Google ❤️ Open Source for everyone.
-
Alibaba
Alibaba Open Source for everyone
-
D3
Data-Driven Documents codes.
-
Tencent
China tencent open source team.
from curl.