search

OpenBMB / ModelCenter

Efficient, Low-Resource, Distributed transformer implementation based on BMTrain
https://modelcenter.readthedocs.io
Apache License 2.0
233 stars 28 forks source link

[BUG] Error in the demonstration code #36

Closed Yangyi-Chen closed 1 year ago

Yangyi-Chen commented 1 year ago

Describe the bug Error in the demonstration code

I run the demonstration code in README. In 4. Training the model: optim_manager = bmt.optim.OptimManager(loss_scale=1024) The error message: AttributeError: module 'bmtrain.optim' has no attribute 'OptimManager'

Could you help look into this problem, many thanks!