[+] add updates in training

update bu and bi and params