Comments (3)
Hi, @hytseng0509
I think they have stated clearly about the GPU memory requirement in the appendix part of their paper. Namely, two RTX 6000 GPUs or a single RTX 8000 GPU.
from pi-gan.
Hi!
I tried the settings below for CelebA (I used 2080Ti and it didn't went OOM ):
0: {'batch_size': 40, 'num_steps': 12, 'img_size': 32, 'batch_split': 2, 'gen_lr': 6e-5, 'disc_lr': 2e-4},
int(10000): {'batch_size': 20, 'num_steps': 12, 'img_size': 64, 'batch_split': 4, 'gen_lr': 3e-5, 'disc_lr': 1e-4},
int(50000): {'batch_size': 8, 'num_steps': 12, 'img_size': 128, 'batch_split': 8, 'gen_lr': 1e-5, 'disc_lr': 5e-5},
int(200000): {},
I think that we don't have to go through all the training steps(200000), which is too time consuming. It was observed that when I merely trained the model for about 1h, it could generate not bad images.
PS: This set means that we train 10000 steps, 40000 steps and 150000 steps for 32x32, 64x64, 128x128 respectively. With this set, it will take about 9h to finish the first two stages(32x32 and 64x64) with a 2080Ti.
from pi-gan.
I have the same question. Could you provide the curriculum file for CelebA that produces the number in the paper?
Also, whats the GPU memory requirement for that training? Thanks!
from pi-gan.
Related Issues (20)
- generator.losses can not visual using tensorboard
- Can't pickle weakref objects HOT 7
- Reproducing the shape results on the project page
- Inverse rendering script assumes strictly oriented input image
- Wrong dataset_path for CARLA and CATS
- hello,how to get the 3d presentation as you mentioned in fig.9 by projecting the depth map? thanks
- Query about Training Time HOT 19
- generated images colours HOT 1
- Intuition behind scaling frequency output of mapping network?
- How many pictures did you use to calculate the FID value of different datasets in the paper? HOT 1
- the link to the datasets HOT 1
- Error when Rendering Videos HOT 4
- 3d mesh reconstruction quality
- Inverse code can't render fine detail as in demo
- Use the same z for training G and D in each iteration.
- Out of memory by training with several 1080ti
- Why I cannot load the pretrained model meeting 'no module named torch_ema'? HOT 1
- Does the pi-GAN discriminator receive the camera pose during training?
- About the lr for G and D
- Dataset cannot be found
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 pi-gan.