8 Commits

Author	SHA1	Message	Date
gerazov	af46727517	loading last checkpoint/best_model works, deleting last best models options added, loading last best_loss added	2021-02-12 02:12:00 +01:00
erogol	c65712426a	change noise scheduling for wavegrad. Compute beta values externally to enable better flexibility	2020-11-14 13:01:10 +01:00
erogol	a44ef58aea	wavegrad weight norm refactoring	2020-10-30 13:23:24 +01:00
erogol	14c2381207	weight norm and torch based amp training for wavegrad	2020-10-29 12:31:43 +01:00
erogol	5b5b9fcfdd	wavegrad config updates	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