ColossalAI
                                
                                
                                
                                    ColossalAI copied to clipboard
                            
                            
                            
                        [BUG]: chat optimizer.backward(loss)
π Describe the bug
ColossalAI/applications/Chat/coati/trainer/strategies/colossalai.py file 150 line optimizer.backward(loss) should be a code error
Environment
No response
Bot detected the issue body's language is not English, translate it automatically. π―ππ»π§βπ€βπ§π«π§πΏβπ€βπ§π»π©πΎβπ€βπ¨πΏπ¬πΏ
Title: [BUG]:
Does it fail anyway?
Does it fail anyway?
I'm not sure, but I didn't find the definition of the optimizer.backward(loss) function, and there is no such usage in the original pytorch
Hi, we have adapted the optimizer API for this use!