Comments (3)
Recent experiment shows that model with gradient checkpointing and SGD optim fits into 16Gb of GPU RAM.
Try this model initialization:
model = GPT2LMHeadModel.from_pretrained("sberbank-ai/mGPT", gradient_checkpointing=True, use_cache=False)
, but remove further conversion model = model.half()
For me it consumes 15774Mb of RAM with batch size 1, seq length 512.
from mgpt.
from mgpt.
Hi! Unfortunately, the amount of GPU memory of colab instances is very limited, so to be able to fit tuning example in it we convert model to an fp16 there and use SGD optimizer that not requires additional memory for parameters. But this tuning mode is definitely far from stable and can lead to a NaN loss due to limitations of fp16 numeric range. We're planning to make an updated version of colab tuning with more advanced techniques to reduce memory consumption. This could possibly allows us to use mixed-precision training with this memory constrains.
For now you can try to tune model using notebook from this repo on your own GPU, it uses float32 training and AdamW optimizer but requires 32Gb of GPU RAM.
from mgpt.
Related Issues (13)
- Cannot run the generation example notebook - which transformers version to use? HOT 1
- Can you share the script for training your tokenizer?
- Can you share the data stat?
- 1.3B model HOT 1
- add web demo to Huggingface HOT 3
- Should we add "<|endoftext|>" explicitly to the prompt? HOT 4
- Batch size increase HOT 3
- mGPT 13B parameters HOT 1
- Non-English prompts
- add new language to the model? HOT 2
- Fine-tunning with specific words for non english language
- Model.zip comes back as not an archive 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 mgpt.