ColossalAI-Examples
ColossalAI-Examples copied to clipboard
Python Exception when running BERT Examples
🐛 Describe the bug
When running BERT sequence example with README, an exception happened.
File "train.py", line 240, in main grad_norm = grad_norm.item() AttributeErrorAttributeError: : 'float' object has no attribute 'item''float' object has no attribute 'item'AttributeError
After annotating line 240, it works.
Environment
No response