You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

Zhen Wang be3777a50a Add pure fp16 training with master weights. (#27712) ... * add the weight decay func for the momentum op * Add the multi_precision function in Momentum Optimizer. * Make sure that the initial value of master weights are same with the fp16 weights. * add static loss scaling. * add the rescale_grad function in the pure fp16 training. * use the original momentum updating method. * Polish some codes, such as variable names. * add docstring for apis. * update the var creation details of _create_master_weight. * not modify codes about imperative momentum updating. * Fix the error of test_dist_sparse_tensor_load_momentum UT. * add unit test for multi precision fp16 training. * add more unit tests for CI. * Use lower threshold values for allclose comparing in test_multi_precision_fp16_train UT. * For CI Coverage Checking.		5 years ago
..
fluid	Add pure fp16 training with master weights. (#27712)	5 years ago
scripts	add compile option WITH_TENSORRT (#29208)	5 years ago
testing	Fix gpu memory allocation bug. (#28703)	5 years ago
.common_test_util.sh	fix spelling errors (#17941)	6 years ago
.gitignore	…
.set_port.sh	revert the modified shell script (#28453)	5 years ago
.set_python_path.sh	revert the modified shell script (#28453)	5 years ago
CMakeLists.txt	…
http.log	Support load state dict form `inference model` format save result (#26718)	5 years ago