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 |