12 Commits

Author	SHA1	Message	Date
Daniel Povey	b0f0c6c4ab	Setting lr_update_period=(200,4k) in train.py	2022-07-25 04:38:12 +08:00
Daniel Povey	25cb8308d5	Add max_block_size=512 to PrAdam	2022-07-12 08:35:14 +08:00
Daniel Povey	50ee414486	Fix train.py for new optimizer	2022-07-09 10:09:53 +08:00
Daniel Povey	e996f7f371	First version I am running, of the speedup code	2022-06-18 15:17:51 +08:00
Daniel Povey	ea5cd69e3b	Possibly fix bug RE learning rate	2022-06-17 20:50:00 +08:00
Daniel Povey	c1f487e36d	Move optim2.py to optim.py; use this optimizer in train.py	2022-06-13 16:05:46 +08:00
Daniel Povey	a9a172aa69	Multiply lr by 10; simplify Cain.	2022-06-04 15:48:33 +08:00
Daniel Povey	ca09b9798f	Remove decomposition code from checkpoint.py; restore double precision model_avg	2022-06-01 14:01:58 +08:00
Daniel Povey	b2259184b5	Use single precision for model average; increase average-period to 200.	2022-05-31 14:31:46 +08:00
Daniel Povey	ab9eb0d52c	Use decompose=True arg for model averaging	2022-05-31 14:28:53 +08:00
Daniel Povey	1651fe0d42	Merge changes from pruned_transducer_stateless4->5	2022-05-31 13:00:11 +08:00
Daniel Povey	741dcd1d6d	Move pruned_transducer_stateless4 to pruned_transducer_stateless7	2022-05-31 12:45:28 +08:00