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 |