Closed Kairixir closed 1 year ago
create and train transformer model add GPU functionality
Transformer part of code changes train. If you use train after this section it might invoke unexpected behaviour.
Check out this pull request on
See visual diffs & provide feedback on Jupyter Notebooks.
Powered by ReviewNB
create and train transformer model add GPU functionality
Transformer part of code changes train. If you use train after this section it might invoke unexpected behaviour.