ERNIE2.0 run_classifier.py fp16 error
Created by: yangze01
Traceback (most recent call last): File "run_classifier.py", line 405, in main(args) File "run_classifier.py", line 126, in main use_fp16=args.use_fp16) File "/home/yaze/remote/ERNIE-develop/optimization.py", line 117, in optimization param_grads, train_program, startup_prog, loss_scaling) File "/home/yaze/remote/ERNIE-develop/utils/fp16.py", line 85, in create_master_params_grads master_grad = master_grad / float(loss_scaling) TypeError: float() argument must be a string or a number, not 'Variable'