5 Commits

Author	SHA1	Message	Date
erogol	c8a4c771a8	train wavegrad updates	2020-10-29 12:31:43 +01:00
erogol	f79bbbbd00	use Adam for wavegras instead of RAdam	2020-10-29 12:31:43 +01:00
erogol	7bcdb7ac35	wavegrad updates	2020-10-29 12:31:43 +01:00
erogol	a1582a0e12	fix distributed training for train_* scripts	2020-10-29 12:31:43 +01:00
erogol	e02cd6a220	initial wavegrad layers model and trainig script	2020-10-29 12:30:37 +01:00