ColossalAI-Examples icon indicating copy to clipboard operation
ColossalAI-Examples copied to clipboard

Python Exception when running BERT Examples

Open Wesley-Jzy opened this issue 3 years ago • 0 comments
trafficstars

🐛 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

Wesley-Jzy avatar Mar 15 '22 09:03 Wesley-Jzy