Comments (3)
Hi, I open the testing function during training. Normally, the testing function will load the image with its original size. Say you have an image over 1000x1000, then the output will be 2000x2000. Therefore, it might take up a lot of GPU space. I think this could be a cause leading to your problem. You can try to comment this line: test_func(opt, webpage, epoch=str(epoch))
in train.py to avoid the testing process during training or try a bit smaller image. You can also try to modify the options about loading image to a smaller size rather than the original size (in test_function.py).
from non-stationary_texture_syn.
Thanks for the response! Turned out I need to change the volatile flag to
with torch.no_grad():
self.real_A = Variable(self.fake_B.data)
self.fake_B = self.netG.forward(self.real_A)
because the volatile flag is deprecated in Torch 0.4.0. After this change, I was able to generate 2400x2400 image using a 1080 card.
Now I wonder how much extra juice can we squeeze out of it. Do you think it's possible to generate 4K image?
from non-stationary_texture_syn.
I'm not quite sure about space usage. My experiments are mostly conducted on one Titan Xp card with 12GB GPU memories. My most training images are around 800~.
from non-stationary_texture_syn.
Related Issues (17)
- Could you please provide the trained model? HOT 1
- No pattern expansion
- Excuse me, but do you have a larger train dataset? HOT 1
- Visualization of the intermediate result of residual blocks
- The usage of the style loss
- about the style_loss backward HOT 1
- Pretrained models HOT 4
- Trained model
- vgg_conv.pth not available HOT 1
- Changes to work on pytorch 3.9.0 and cuda 11.3
- Quick Question HOT 1
- backward_G in half_gan_style.py HOT 2
- errors caused by network connection HOT 5
- Generalization ability of model HOT 4
- About experiments in your paper HOT 2
- Augmentation of diversity by shuffling tiles HOT 2
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 non-stationary_texture_syn.