grad_norms.append(g.to(get_accelerator().device_name(), non_blocking=True).float.norm(2)) 这里的float后面是不是缺了一个()导致代码执行报错
可以麻烦您提供一下您这边执行有bug的demo程序么?我们内部核对一下;
可以麻烦您提供一下您这边执行有bug的demo程序么?我们内部核对一下;
如果终端输出的bug调用栈明确指向这里,可以试下按您这边的结论修改一下进行对比呢。