Comments (4)
Thanks!
The VQGAN benefits greatly from training it as long as possible (provided the data set is large enough and overfitting is a secondary concern), and tuning in the discriminator rather late. When training on ImageNet, for example, I would recommend 3-5epochs without the adversarial loss (but more is better) and then training for at least another 3-5 epochs with the discriminator turned on (again, more=better).
The stopping condition for the transformer is usually when it starts overfitting in terms of NLL on held-out test data.
from taming-transformers.
May I ask how long it usually takes to train on the ImageNet and how many GPUs are used?
from taming-transformers.
Any updates on training times? Costs?
from taming-transformers.
mark
from taming-transformers.
Related Issues (20)
- How to do image inpainting with vqgan?
- How to train the transformer after training the vqgan model? HOT 9
- If my condition is class label, how to set config file
- Cannot use gpu to fine-tune on custom-data HOT 2
- gpu training hangs HOT 2
- Where is the model trained on my custom data? HOT 1
- When I train, the code is getting slower and slower, is it normal? HOT 1
- README Error -- yaml path
- Dockerfile/environment.yaml for CUDA Version 12.3? HOT 5
- 'VQModel' object has no attribute 'transformer'
- How to train your own Transformer
- How to distinguish sos token(default = 0) and quantified image token zero ?
- How to use the sliding attention window mechanism? HOT 3
- Is there a plan to upgrade this repo to Pytorch 2.0
- ModuleNotFoundError: No module named 'cutlit'
- ncc HOT 1
- Confusion about the dimension of latent vectors
- generate white images during training HOT 1
- : Parameter indices which did not receive grad for rank 0: 338 339 340 341 342 343 344 345 346 347 348 349 350 HOT 3
- A question regarding the embed_dim parameter in the config file.
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 taming-transformers.