Comments (2)
If you have a GPU, it is not big enough to store the model + feature maps + output. You can see that the OOM happens on the final layer, when the output has the size 4800x4800x32. You can try to do inference on smaller images, or try splitting the model over multiple GPUs.
If you have more RAM than your GPU does, you can try doing inference on CPU (which will use RAM for allocating tensors). It will be slow but at least it will work. You can see here how to do this: link
from fast-srgan.
I see. Thanks for replying so quickly.
from fast-srgan.
Related Issues (20)
- How to predict a single image HOT 1
- Image was white until I made this change HOT 2
- is there any plan to port the SR model onto a mobile phone? HOT 1
- feeding Low Res images to Generator without downsampling High Res images HOT 1
- infer.py not working in the same dependencies... HOT 1
- Training for Higher Upsampling Levels HOT 2
- ValueError: Input 0 is incompatible with layer model_2: expected shape=(None, 96, 96, 3), found shape=(None, 480, 640, 3) from infer.py HOT 9
- pre-trained model D (Discriminator) not found HOT 1
- VGG19 for grayscale images HOT 1
- how to process sr in real time speed for developing video cam HOT 1
- Models for higher resolution HOT 1
- Any plans making ML Upscaler for gaming/3D? HOT 1
- Running the model in Real time with webcam HOT 1
- logs HOT 1
- SRGAN beginner HOT 6
- Incompatible shape HOT 4
- ValueError: Exception encountered when calling layer "model" (type Functional) HOT 4
- Input 0 of layer "model_2" is incompatible with the layer: expected shape=(None, 96, 96, 3), found shape=(None, 384, 384, 3) HOT 1
- Training requirements? HOT 1
- The model file does not match your network HOT 1
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 fast-srgan.