5 Commits

Author	SHA1	Message	Date
yaozengwei	a72048be3e	remove device convertion for model_avg	2022-05-06 15:58:52 +08:00
yaozengwei	3c405fb6fd	minor fix	2022-05-06 15:37:30 +08:00
yaozengwei	ae50acad8b	explicitly convert model_avg to cpu	2022-05-06 15:35:09 +08:00
yaozengwei	c3bb03253f	keep model_avg on cpu	2022-05-06 14:45:00 +08:00
Zengwei Yao	00c48ec1f3	Model average (#344) ... * First upload of model average codes. * minor fix * update decode file * update .flake8 * rename pruned_transducer_stateless3 to pruned_transducer_stateless4 * change epoch number counter starting from 1 instead of 0 * minor fix of pruned_transducer_stateless4/train.py * refactor the checkpoint.py * minor fix, update docs, and modify the epoch number to count from 1 in the pruned_transducer_stateless4/decode.py * update author info * add docs of the scaling in function average_checkpoints_with_averaged_model	2022-05-05 21:20:04 +08:00