tkgix
|
aa24932c51
|
Update main.py
Simple solution to a common problem with state_dict
|
2022-12-10 22:10:41 +09:00 |
|
Victor Hall
|
02487c8db8
|
work on next release
|
2022-11-13 21:45:51 -05:00 |
|
Victor Hall
|
aaa732007a
|
ok emergency over with multiaspect
|
2022-11-10 18:29:31 -05:00 |
|
Victor Hall
|
e7a7b33c1e
|
autoprune all script
|
2022-11-09 01:34:47 -05:00 |
|
Victor Hall
|
853a7500f4
|
some final cleanup and readme updates
|
2022-11-06 19:59:37 -05:00 |
|
Victor Hall
|
b7779591b1
|
multi aspect ratio support
|
2022-11-05 11:41:48 -04:00 |
|
Victor Hall
|
89cc295515
|
multi-aspect image support work
|
2022-11-03 19:47:54 -04:00 |
|
Victor Hall
|
93bbd30caa
|
cleaning up training process
|
2022-11-02 22:23:09 -04:00 |
|
Victor Hall
|
30f78de976
|
fix batch size hunting at start of training
|
2022-10-28 21:37:21 -04:00 |
|
Victor Hall
|
f5aba7293d
|
big readme update, cruft and dead code removed, reverting some stuff back to xavier og
|
2022-10-25 22:45:47 -04:00 |
|
Victor Hall
|
09d67853f0
|
trainer ymls and fix main
|
2022-10-24 22:58:36 -04:00 |
|
Victor Hall
|
2026137e46
|
removing cruft
|
2022-10-23 19:48:39 -04:00 |
|
Victor Hall
|
f8fdd54485
|
main.py cleanup
|
2022-10-22 17:13:51 -04:00 |
|
Kane Wallmann
|
e2462ba2dc
|
Change dataloader to correctly pair training and regularization images
|
2022-10-01 13:56:27 +10:00 |
|
Kane Wallmann
|
648b87a200
|
Modifications to support multiple identifier training
|
2022-10-01 10:53:24 +10:00 |
|
David Bielejeski
|
0d60899e04
|
Re-enabling logging of images.
|
2022-09-30 15:25:15 -05:00 |
|
David Bielejeski
|
40eb005c3d
|
Fixing log_img blowing up.
|
2022-09-29 18:35:40 -05:00 |
|
David Bielejeski
|
ac9e6bac9c
|
Actually fixing max_steps error.
|
2022-09-28 16:27:18 -05:00 |
|
David Bielejeski
|
aadc5ebea0
|
Fixing for duplicate arg error.
|
2022-09-28 16:17:25 -05:00 |
|
David Bielejeski
|
fce2d1f17d
|
Fixing batch_size to be max_steps. Turning --no-test on by default.
|
2022-09-28 15:19:31 -05:00 |
|
David Bielejeski
|
a524cfb751
|
Not requiring the batch size
|
2022-09-26 18:13:12 -05:00 |
|
David Bielejeski
|
ee5b27386f
|
Matching main for ddp
|
2022-09-26 12:44:22 -05:00 |
|
David Bielejeski
|
67154ad9b5
|
Minor updates
|
2022-09-26 12:41:03 -05:00 |
|
David Bielejeski
|
2671369079
|
Attempting auto-prune and checkpoint saving
|
2022-09-26 08:27:10 -05:00 |
|
David Bielejeski
|
fd298537c2
|
Porting changes over from my branch.
|
2022-09-26 08:01:18 -05:00 |
|
JoePenna
|
3d97f68047
|
ddpm changes
|
2022-09-25 22:23:08 -07:00 |
|
Christopher Gammage
|
25e5c2520e
|
get rid of pycache stuff and make work with windows
|
2022-09-25 17:09:41 -07:00 |
|
JoePenna
|
33be8bb255
|
Adding pruning option
Adding a version of the main.py script that automatically prunes about 10GB off the ckpt.
prune.py is from waifu diffusion repo
|
2022-09-20 13:39:57 -07:00 |
|
XavierXiao
|
8f22429d74
|
initial commit
|
2022-09-06 00:00:21 -07:00 |