475 Commits

Author	SHA1	Message	Date
Kohya S	9ff26af68b	Update to add grad_ckpting etc to metadata	2023-01-21 12:36:31 +09:00
Kohya S	7dbcef745a	Merge pull request #77 from space-nuko/ss-extra-metadata ... More helpful metadata	2023-01-21 12:18:23 +09:00
Kohya S	758323532b	add save_last_n_epochs_state to train_network	2023-01-19 20:59:45 +09:00
space-nuko	da48f74e7b	Add new version model/VAE hash to training metadata	2023-01-18 23:00:16 -08:00
michaelgzhang	303c3410e2	expand details in tensorboard logs ... - Update tensorboard logging to track both unet and textencoder learning rates - Update tensorboard logging to track both current and moving average epoch loss - Clean up tensorboard log variable names for dashboard formatting	2023-01-18 13:10:13 -06:00
space-nuko	de1dde1a06	More helpful metadata ... - dataset/reg image dirs - random session ID - keep_tokens - training date - output name	2023-01-17 16:28:35 -08:00
Yuta Hayashibe	3eb8fb1875	Make not to save state when args.save_state is False	2023-01-18 01:31:38 +09:00
Yuta Hayashibe	3815b82bef	Removed --save_last_n_epochs_model	2023-01-16 21:02:27 +09:00
Yuta Hayashibe	c6e28faa57	Save state when args.save_last_n_epochs_state is designated	2023-01-15 19:43:37 +09:00
Yuta Hayashibe	a888223869	Fix a bug	2023-01-15 18:02:17 +09:00
Yuta Hayashibe	d30ea7966d	Updated help	2023-01-15 18:00:51 +09:00
Yuta Hayashibe	df9cb2f11c	Add --save_last_n_epochs_model and --save_last_n_epochs_state	2023-01-15 17:52:22 +09:00
Kohya S	186a2665ad	Merge branch 'main' into textual_inversion	2023-01-15 16:08:53 +09:00
Kohya S	aa40cb9345	Add train epochs and max workers option to train	2023-01-15 13:07:47 +09:00
Kohya S	c1b14fcdd6	initial version of TI	2023-01-12 20:47:08 +09:00
Kohya S	e4f9b2b715	Add VAE to meatada, add no_metadata option	2023-01-11 23:12:18 +09:00
space-nuko	2e4ce0fdff	Add training metadata to output LoRA model	2023-01-10 02:49:52 -08:00
Kohya S	673f9ced47	Fix '*' is not working for DreamBooth	2023-01-09 21:06:58 +09:00
Gaetano Bonofiglio	d8da85b38b	fix file not found when [ is in the filename	2023-01-09 11:40:00 +01:00
Kohya S	6b62c44022	fix errors in fine tuning	2023-01-08 21:40:40 +09:00
Kohya S	1945fa186d	Show error if caption isn't UTF-8, add bmp support	2023-01-08 18:50:52 +09:00
Kohya S	9f1d3aca24	add save_state_on_train end, fix reg imgs repeats	2023-01-07 20:20:37 +09:00
Kohya S	f56988b252	unify dataset and save functions	2023-01-05 08:10:22 +09:00
Kohya S	4c35006731	split common function from train_network to util	2023-01-03 20:22:25 +09:00
Kohya S	6b522b34c1	move code for xformers to train_util	2023-01-02 16:08:21 +09:00