Coding a Transformer from scratch on PyTorch, with full explanation, training and inference.

Рет қаралды 152,671

Күн бұрын

In this video I teach how to code a Transformer model from scratch using PyTorch. I highly recommend watching my previous video to understand the underlying concepts, but I will also rehearse them in this video again while coding. All of the code is mine, except for the attention visualization function to plot the chart, which I have found online at the Harvard university's website.
Paper: Attention is all you need - arxiv.org/abs/1706.03762
The full code is available on GitHub: github.com/hkproj/pytorch-tra...
It also includes a Colab Notebook so you can train the model directly on Colab.
Chapters
00:00:00 - Introduction
00:01:20 - Input Embeddings
00:04:56 - Positional Encodings
00:13:30 - Layer Normalization
00:18:12 - Feed Forward
00:21:43 - Multi-Head Attention
00:42:41 - Residual Connection
00:44:50 - Encoder
00:51:52 - Decoder
00:59:20 - Linear Layer
01:01:25 - Transformer
01:17:00 - Task overview
01:18:42 - Tokenizer
01:31:35 - Dataset
01:55:25 - Training loop
02:20:05 - Validation loop
02:41:30 - Attention visualization

Пікірлер: 297

@comedyman4896 9 ай бұрын

personally, I find that seeing someone actually code something from scratch is the best way to get a basic understanding

@zhilinwang6303 5 ай бұрын

indeed

@user-jb2ex2ux9i 5 ай бұрын

indeed

@CM-mo7mv 4 ай бұрын

i don't need to see someone typing... but you might also enjoy watching the gras grow or paint dry

@FireFly969 2 ай бұрын

Yeah, and you see how these technologies works. It's insane, that in the end it looks easy that you can do something that companies of millions and billions of dollars do. In a small way but the same idea at the end.

@maskedvillainai 2 ай бұрын

Yeah kinda ironic how that works. The simplest stuff required the most complex explanations

@faiyazahmad2869 2 күн бұрын

One of the best tutorial to understand and implement the Transformer model...Thank you for making such a wonderful video

@umarjamilai Жыл бұрын

The full code is available on GitHub: github.com/hkproj/pytorch-transformer It also includes a Colab Notebook so you can train the model directly on Colab. Of course nobody reinvents the wheel, so I have watched many resources about the transformer to learn how to code it. All of the code is written by me from zero except for the code to visualize the attention, which I have taken from the Harvard NLP group article about the Transformer. I highly recommend all of you to do the same: watch my video and try to code your own version of the Transformer... that's the best way to learn it. Another suggestion I can give is to download my git repo, run it on your computer while debugging the training and inference line by line, while trying to guess the tensor size at each step. This will make sure you understand all the operations. Plus, if some operation was not clear to you, you can just watch the variables in real time to understand the shapes involved. Have a wonderful day!

@AiEdgar Жыл бұрын

The best video ever

@odyssey0167 9 ай бұрын

Can you provide with the pretrained models?

@wilfredomartel7781 3 ай бұрын

🎉is this Bert architecture?

@yangrichard7874 6 ай бұрын

Greeting from China! I am PhD student focused on AI study. Your video really helped me a lot. Thank you so much and hope you enjoy your life in China.

@umarjamilai 6 ай бұрын

谢谢你！我们在领英联系吧

@ArslanmZahid 7 ай бұрын

I have browsed KZfaq for the perfect set of videos on transformer, but your set of videos (the video explanation you did on the transformer architecture) and this one is by far the best !! Take a bow brother, you have really contributed to the viewers in amount you cant even imagine. Really appreciate this !!!

@aiden3085 7 ай бұрын

Thank you Umar for our extraordinary excellent work! Best transformer tutorial ever I have seen!

@zhengwang1402 7 ай бұрын

This feels really fantastic when looking someone write a program from bottom up

@kozer1986 8 ай бұрын

I'm not sure if it is because I have study this content 1000000 times or not, but is the first time that I understood the code, and feel confident about it. Thanks!

@abdullahahsan3859 8 ай бұрын

Keep doing what you are doing. I really appreciate you taking out so much time to spread such knowledge for free. Been studying transformers for a long time but never have I understood it so well. The theoretical explanation in the other video combined with this practical implementation, just splendid. Will be going through your other tutorials as well. I know how much time taking it is to produce such high level content and all I can really say is that I really am grateful for what you are doing and hope that you continue doing it. Wish you a great day!

@umarjamilai 8 ай бұрын

Thank you for your kind words. I wish you a wonderful day and success for your journey in deep learning!

@MuhammadArshad 8 ай бұрын

Thank God, it's not one of those 'ML in 5 lines of Python code' or 'learn AI in 5 minutes'. Thank you. I can not imagine how much time you must have spent on making this tutorial. thank you so much. I have watched it three times already and wrote the code while watching the second time (with a lot of typos :D).

@abdulkarimasif6457 Жыл бұрын

Dear Umar, your video is full of knowledge; thanks for sharing.

@raviparihar3298 Ай бұрын

best video I have ever seen on whole youtube eon transformer model. Thank you so much sir!

@maxmustermann1066 9 ай бұрын

This video is incredible, never understood it like this before. I will watch your next videos for sure, thank you so much!

@sagarpadhiyar3666 2 ай бұрын

Best video I came across for transformer from scratch.

@shresthsomya7419 5 ай бұрын

Thanks a lot for such a detailed video. Your videos on transformer are best.

@manishsharma2211 7 ай бұрын

WOW WOW WOW, though it was a bit tough for me to understand it, I was able to understand around 80 % of the code, beautiful. Thank you soo much

@michaelscheinfeild9768 10 ай бұрын

Im enjoying clear explanation of The Transformer Coding !

@mikehoops 8 ай бұрын

Just to repeat what everyone else is saying here - many thanks for an amazing explanation! Looking forward to more of your videos.

@shakewingo3216 8 ай бұрын

Thanks for making it so easy to understand. I definitely learn a lot and gain much more confidence from this!

@terryliu3635 Ай бұрын

I learnt a lot from following the steps out of this video and create a transformer myself step by step!! Thank you!!

@VishnuVardhan-sx6bq 6 ай бұрын

This is such a great work, I don't really know how to thank you but this is an amazing explanation of an advanced topic such as transformer.

@salmagamal5676 6 ай бұрын

I can't possibly thank you enough for this incredibly informative video

@user-db8nb5wz2z 8 ай бұрын

Really great explanation to understand Transformer, many thanks to you.

@goldentime11 2 ай бұрын

Thanks for your detailed tutorial. Learned a lot!

@user-ru4nb8tk6f 8 ай бұрын

You are a great professional, thanks a ton for this

@linyang9536 6 ай бұрын

这是我见过最详细的从零创建Transformer模型的视频，从代码实现到数据处理，再到可视化，up主真是嚼碎磨细了讲，感谢！

@decarteao 5 ай бұрын

Nn entendi nada! Mas botei meu like.

@astrolillo 3 ай бұрын

@@decarteaoO cara da China e muito engracado con o video

@prajolshrestha9686 8 ай бұрын

I appreciate you for this explanation. Great video!

@balajip5030 8 ай бұрын

Thanks Bro. With your explanation, I am able to build the transformer model for my application. You explained so awesome. Please do what you are doing.

@physicswithbilalasmatullah 3 ай бұрын

Hi Umar. I am a first year student at MIT who wants to do AI startups. Your explanation and comments during coding were really helpful. After spending about 10 hours on the video, I walk away with great learnings and great inspiration. Thank you so much, you are an amazing teacher!

@umarjamilai 3 ай бұрын

Best of luck with your studies and thank you for your support!

@keflatspiral4633 6 ай бұрын

what to say.. just WOW! thank you so much !!

@qikangdeng1487 2 ай бұрын

What a WONDERFUL example of transformer! I am Chinese and I am doing my PhD program in Korea. My research is also about AI. This video helps me a lot. Thank you! BTW, your Chinese is very good!😁😁

@lyte69 8 ай бұрын

Hey there! I enjoyed watching that video, you did a wonderful job explaining everything, and I found it super easy to follow along. Overall, it was a really great experience!

@ghabcdef 5 ай бұрын

Thanks a ton for making this video and all your other videos. Incredibly useful.

@umarjamilai 5 ай бұрын

Thanks for your support!

@saziedhassan3976 10 ай бұрын

Thank you so much for taking the time to code and explain the transformer model in such detail. You are amazing and please do a series on how transformers can be used for time series anomaly detection and forecasting!

@amiralioghli8622 9 ай бұрын

My question and request is same as you. if you found any tutorial please share with me.

@godswillanosike896 3 ай бұрын

Great explanation! Thanks very much

@mohamednabil374 Жыл бұрын

Thanks Umar for this comprehensive tutorial, after watching many videos I would say, this is AWESOME! It would be really nice if you can provide us with more tutorials on Transformers especially training them for longer sequences. :)

@umarjamilai Жыл бұрын

Hi mohamednabil374, stay tuned for my next video on the LongNet, a new transformer architecture that can scale up to 1 billion tokens.

@texwiller7577 3 ай бұрын

Dottore...sei un grande!

@toxicbisht4344 6 ай бұрын

Amazing explanation Thank you for this

@divyanshbansal2321 5 ай бұрын

Thank you mate. You are a godsend!

@skirazai7591 6 ай бұрын

Great video, you are insanely talented btw.

@ansonlau7040 3 ай бұрын

Big thankyou for the video, makes transformer so easy to learn(also the explanation video)👍👍

@JohnSmith-he5xg 8 ай бұрын

Loving this video (only 13 minutes in), really like you using type hints, commenting, descriptive variable names, etc. Way better coding practices than most of the ML code I've looked at. At 13:00, for the 2nd arg of the array indexing, you could just do ":" and it would be identical.

@tonyt1343 6 ай бұрын

Thank you for this comment! I'm coding along with this video and I wasn't sure if my understanding was correct. I'm glad someone else was thinking the same thing. Just to be clear, I am VERY THANKFUL for this video and am in no way complaining. I just wanted to make sure I understand because I want to fully internalize this information.

@oborderies 7 ай бұрын

Sincere congratulations for this fine and very useful tutorial ! Much appreciated 👏🏻

@dapostop7384 Ай бұрын

Wow super usefull! Coding really helps me understand the process better than visuals.

@nhutminh1552 6 ай бұрын

Thank you admin. Your video is great. It helps me understand. Thank you very much.

@MrSupron00 9 ай бұрын

This is excellent! Thank you for putting this together. I do have one point of confusion with how the final multihead attention concatenation takes place. I believe the concatenation takes place on line 110 where V' = (V1, V2,.. Vh) (sequenc_length, h*dk) This is intended to be multiplied by matrix W0 (h*dk, dmodel) to give something of shape (sequenc_length, dmodel ) as is required. However, here you implement a linear layer operation which takes the concat V' (sequence_length, d_model) and is fed into a linear layer constructed so that we do the following: W*V'+b where the dimension of W and b are chose to satisfy the output dimension. This is different from multiplying directly with a predefined trainable matrix of size W0. Now, I can see how these are nearly the same thing and in practice it may not matter, but it would be helpful to point out these tricks of the trade so folks like myself don't get bogged down with these subtleties. Thanks

@codevacaphe3763 Ай бұрын

Hi, I just happen to see your video. It's really amazing, your channel is so good with valuable information. Hope, you keep this up because I really love your contents.

@Mostafa-cv8jc 7 ай бұрын

Very good video. Tysm for making this, you are making a difference

@si0n4ra 11 ай бұрын

Umar, thank you for the amazing example and clear explanation of all your steps and actions.

@umarjamilai 11 ай бұрын

Thank you for watching my video and your kind words! Subscribe for more videos coming soon!

@si0n4ra 11 ай бұрын

@@umarjamilai , mission completed 😎. Already subscribed. All the best, Umar

@forresthu6204 8 ай бұрын

At 22:39, it describes the essentials of self-attentions computation in very clear and easy to understand way.

@amiralioghli8622 9 ай бұрын

Thank you so much for taking the time to code and explain the transformer model in such detail. You are amazing and, if possible please do a series on how transformers can be used for time series anomaly detection and forecasting. it is extremly necessary on yotube for somone! Thanks in advance.

@jeremyregamey495 7 ай бұрын

I love your videos. Thank you for sharing your knowledge and i cant wait to learn more.

@gunnvant 9 ай бұрын

This was really good. I understood multihead attention better with the code explanation.

@user-eu3ok8dc8b 9 ай бұрын

one of the best videos thanks a lot for the video.

@LeoDaLionEdits Ай бұрын

thank you so much for these videos

@coc2912 10 ай бұрын

Thanks for your video and code.

@cicerochen313 8 ай бұрын

Awesome! Highly appreciate. 超級讚！非常的感謝。

@user-ul2mw6fu2e 6 ай бұрын

Wow Your explanation amazing

@user-qo7vr3ml4c Ай бұрын

Thank you very much, this is very useful.

@Patrick-wn6uj 3 ай бұрын

Hi Umar thank you for all the work you are doing, please consider making a video like this on vision transformers

@jihyunkim4315 8 ай бұрын

perfect video!! Thank you so much. I always wonder the detail code and its explanation and now I almost understand all of it. thanks:) you are the best for me!

@umarjamilai 8 ай бұрын

You're welcome!

@SyntharaPrime 7 ай бұрын

Great Job. Amazing. Thanks a lot. I really appreciate you. It is so much effort.

@sypen1 7 ай бұрын

This is amazing thank you 🙏

@angelinakoval8360 7 ай бұрын

Dear Umar, thank you so so much for the video! I don't have much experience in deep learning, but your explanations are so clear and detailed I understood almost everything 😄. It wil be a great help for me at my work. Wish you all the best! ❤

@umarjamilai 7 ай бұрын

Thank you for your kind words, @angelinakoval8360!

@shengjiadiao3166 18 сағат бұрын

the contents are crazy !!!!

@michaelscheinfeild9768 10 ай бұрын

i enjoyed the video ! now i can transform the world !

@sypen1 7 ай бұрын

Mate you are a beast!

@pawanmoon 9 ай бұрын

Great work!!

@FailingProject185 4 ай бұрын

You are one of the coolest dude in this area. It'd be helpful if you provide a roadmap to reach your expertise. I'd really love to learn from you but i can't understand. Roadmap will help so many of your subscribers.

@md.shahabulalam9484 2 ай бұрын

Umar thank you so much.........

@user-sp5pf5du3m 4 ай бұрын

You are a genius

@JohnSmith-he5xg 8 ай бұрын

OMG. And you also note Matrix shapes in comments! Beautiful. I actually know the shapes without having to trace some variable backwards.

@omidsa8323 5 ай бұрын

Great Job!

@reannwu3283 9 ай бұрын

This is work of art.

@ngocchienchu109 11 ай бұрын

thank you so much

@user-wr4yl7tx3w 8 ай бұрын

the code is really well written. very easy and nicely organized.

@babaka1850 Ай бұрын

for determining the max len of tgt sentence, I believe you should point to tokenizer_tgt rather than tokenizer_src. tgt_ids = tokenizer_tgt.encode(item['transaltion'][config['ang_tgt']]).ids

@aspboss1973 10 ай бұрын

Its really awesome video with clear explanations. And flow of code is very easy to understand. One question, how to implement this transformer architecture for Question-Answer based model ? (Q/A on very specific topic lets say a manual of instrument..) Thank you ! so much for this video !!!

@phanindraparashar8930 Жыл бұрын

It is really amazing video. I tried understanding the code of it from various other youtube channel; but was always getting confused. Thanks a lot :) . Can you make a series on BERT & GPT aswell; where you build these models and train on custom data?

@umarjamilai Жыл бұрын

Hi Phanindra! I'll definetely continue making more videos. It takes a lot of time and patience to make just one video, not considering the preparation time to study the model, write the code and test it. Please share the channel and subscribe, that's the biggest motivation to continue providing high quality content to you all.

@rubelahmed5458 7 ай бұрын

A coding example for BERT would be great!@@umarjamilai

@user-gj2cl2rr9x 4 ай бұрын

in the 13:13/2:59:23, when we build the PositionalEncoding function, this line x = x + (self.pe[:,:x.shape[1],:]).requires_grad_(False), the x.shape[1] looks like not be used in the transformer model, because when we build the dataset.py function, we pad all the sentences into the same length, and then we load the (batch, seq_len, input_embedding_dim) into the PositionalEncoding function, where all x.shape[1] in the batch is the seq_len, instead of varying by their original sentence length.

@PP-qi9vn 8 ай бұрын

Thanks!

@wellhellothere1785 10 ай бұрын

How do you suggest to apply this model to time series forcasting what do you think should be changed, so far I believe that there is no source language and target language in the forcasting there is only time based. and also for the error or loss function I should use MSE in this case. Is there anything else I might be missing ?

@Hdjandbkwk 10 ай бұрын

Just want to say thank you!! This is easily one of my favorite video on KZfaq! I have watched a few videos on transformers but none explained it as clear as you, at first I was scared by the length of the video but you managed to have my attention for the full 3 hours! Following your instructions I am now able to train my very first transformer! Btw, I am using the tokenizer the way you are but looking at the tokenizer file it looks like my tokenizer didn’t split the sentences into words and it is using the whole sentence as token. Do you have any idea why? I am using mac if that matters.

@umarjamilai 10 ай бұрын

Hi! Thanks for your kind words! Make sure your PreTokenizer is the "Whitespace" one and that the Tokenizer is the "WordLevel" tokenizer. As a last resort, you can clone the repository from my GitHub and compare my code with yours. Have a wonderful rest of the day!

@Hdjandbkwk 10 ай бұрын

I have PreTokenizer set as whitespace and using WordLevel tokenizer and trainer but it will still encode the sentence as a whole. I did a direct swap to use BPE tokenizer and that is correctly encoding the sentences, maybe there is bug in WordLevel tokenizer for macOS. Another question that I have is what determines the max context size for LLMs? Is it the d_model size?@@umarjamilai

@user-gj2cl2rr9x 4 ай бұрын

Really helpful video. I watched it many times. Hope you enjoy your life in China. 龙年大吉

@umarjamilai 4 ай бұрын

谢谢老板的精准扶贫

@ageofkz 4 ай бұрын

At 29:14, the part on multihead attention, we feed each Q,V, K multiply by Wq, Wv, Wk then split them into n heads then dot product and concat them again. But should we not split them first, then apply Wq_h where Wq_h is the weight matrix for the hth query matrix, same for V and K? Because it seems like we just split them, apply attention, then concat?

@therealvortex100 11 ай бұрын

Hello, at 51:16 why do we add normalization in the end of encoder?

@umarjamilai 11 ай бұрын

Hi! About the layer normalization, there are different opinions on where to add it in the model. I suggest you read this paper (arxiv.org/abs/2002.04745) which discusses this issue. Have a nice day!

@tharllesjhoinessilvate200 8 ай бұрын

Video otimo, pena que não entendo o inglês, sou brasileiro, tinha varias duvidas mas com certeza vou aprender ingles depois venho aqui!! parabéns pelo video

@peregudovoleg 6 ай бұрын

Thank you for your straight to the point no bs videos. Good code alongs and commentary. But it looks like positional encoding aren't correct (as per paper). There is power there in: denominator = np.power(10000, 2*i/d). I get it you decided to use exp+log pair for stability, but no mentions of the power gone. And extra layer norm after encoder. As in we "norm + add" (like you defined in the video, instead add & norm as per paper, but you said "lets stick with it", I understand) but than we norm again, after the last encoder block. Like so: layer norm + add (in last encoder block (inside residual connection ) + layer norm again (inside Encoder class) (I could be wrong but it looks like that)

@vigenisayan2343 4 ай бұрын

it was very useful to watch. Question: What books or learning sources you would suggest to learn pytorch deeply. Thanks

@rafabaranowski513 4 ай бұрын

At 1:42:03 you are using SOS special token from source language tokenizer in sentence with target language. Tokenizers are trained on different languages so is it correct to use special tokens between them? SOS token from source language tokenizer won't have different idx compared to SOS from target language tokenizer?

@albert4392 9 ай бұрын

This is an excellent video, your explanation is so clear and the live coding helps understanding! Can you give us tips to debug such an huge model? Because it is really hard to make sure the model works well. My tips on debugging is to print out the shape of the tensor in each step, but this only make sure the shape is correct, there may be some logical error I may miss out. Thank you!

@umarjamilai 9 ай бұрын

Hi! I'd love to give a golden rule for debugging models, but unfortunately, it depends highly on the architecture/loss/data itself. One thing that you can do is, before training the model on a big dataset, it is recommended to train it on a very small dataset to make sure everything is working and the model should overfit on the small dataset. For example, if instead of training on many books, you train a LLM on a single book, hopefully it should be able to write sentences from that book, given a prompt. The second most important thing is to validate the model as the training is proceeding to verify that the quality is improving over time. Last but not least, use metrics to decide if the model is going in the right direction and make experiments on hyper parameters to verify assumptions, do not just make assumptions without validating them. When you have a model with billions of parameters, it is difficult to predict patterns, so every assumption must be verified experimentally. Have a nice day!

@RecognizeAndNekroniX 3 ай бұрын

Мне кажется, я впервые ставлю лайк на этой площадке

@dengorange2631 2 ай бұрын

既然你懂中文，那我就用中文感谢你吧！感谢你的精彩视频！

@umarjamilai 2 ай бұрын

不用谢😇

@ebadsayed487 17 күн бұрын

Your video is truly amazing, thanks a lot for this. I want to train this model on Summarization Task so what changes I need to do?

@AyushRaj-nt3ot Ай бұрын

sir, your explanation is just beyond awesome!!! Thank you so much for creating such content. Sir I didn't get the residual connections part. As I am from India, I was working on Indic Languages, so i had to make more code but that's just okay. I just want if you could please help in understanding beam search code, the one which you also gave in the GitHub File. Also, if you could give the code for evaluating the BLEU Score. I'll be really grateful to you. And again, thank you so much for such a comprehensive content. We'd love to see your more videos especially in Generative AI! P.S. : I didn't understand how you wrote it, what I've understood is that we have to take the input of the previous layer and then add with o/p of the same layer and then apply layer norm on that. Basically Add and then LayerNorm. Please help me correct mysefl!

@kailazarov107 9 ай бұрын

Really great video - learned a lot. Your inference notebook is using the dataset batching, but how can you build inference with user typed sentences?

@aomo5293 Жыл бұрын

Have made a example for video classification using transformer

@julianmejiez8006 11 ай бұрын

HI thank you for the videos. I got a question, while tryning process im using (en-fr) dataset but im having an error with the paddings. im having raise ValueError("Sentence is too long").How can i avoid this while training, any thoughts on this?.

@umarjamilai 11 ай бұрын

If I remember correctly, that Exception is raised when the sentence length is too short to accomodate for the sentences present in the data set. Try to increase the sentence length of the model.

@cocohand781 4 ай бұрын

i wander what is difference between using lambda function and using direct function in line 163,164,165 thankyou

@kareemsaid8863 2 ай бұрын

i tried to modify the code bit for code generation task and i am stuck at 1.2 loss what do you think is the problem litterily the same code i just changed abit in the get_ds function and get_sequances function and i treated the src tokenizer and tgt tokenizer as the same tokenizer what is wrong with my changes in your implementation ?