TypeErrorTypeError: : AdamOptimizer.__init__() got an unexpected keyword argument 'scale'AdamOptimizer.__init__() got an unexpected keywo$ d argument 'scale'

[Oran-Ac]

When I try to run the examples :bash gpt/gpt2_test.sh It fails and throws out the following errors:

File "/workspace/BMCook/examples/gpt/gpt2_test.py", line 84, in main
File "/workspace/BMCook/examples/gpt/gpt2_test.py", line 84, in main
optimizer = bmt.optim.AdamOptimizer(model.parameters(), scale=2**20)optimizer = bmt.optim.AdamOptimizer(model.parameters(), scale=2**20)
TypeErrorTypeError: : AdamOptimizer.init() got an unexpected keyword argument 'scale'AdamOptimizer.init() got an unexpected keyword argument 'scale'

The bmtrain version I use is v0.2.1

[zzy14]

Thank you for your feedback. We have addressed this issue with a fix. Please try updating to the latest version, and let us know if you encounter any further problems.