6 Commits

Author	SHA1	Message	Date
anton-l	302b86bd0b	Adapt training to the new UNet API	2022-07-21 11:07:21 +02:00
Anton Lozhkov	76f9b52289	Update the training examples (#102) ... * New unet, gradient accumulation * Save every n epochs * Remove find_unused_params, hooray! * Update examples * Switch to DDPM completely	2022-07-20 19:51:23 +02:00
Anton Lozhkov	33abc79515	Update README.md	2022-06-22 13:52:45 +02:00
anton-l	a2117cb797	add push_to_hub	2022-06-21 10:38:34 +02:00
Manuel Romero	57aba1ef50	Fix output path name	2022-06-15 21:45:49 +02:00
Anton Lozhkov	1112699149	add a training examples doc	2022-06-15 16:51:37 +02:00