Commit Graph

13 Commits

Author SHA1 Message Date
Joel Holdbrooks 326d861a86 Push DLMA into main, pass config to resolve
This patch

* passes the configuration (`argparse.Namespace`) to the resolver,
* pushes the DLMA code into the main function,
* makes DLMA take a `list[ImageTrainItem]` instead of `data_root`,
* makes `EveryDreamBatch` take `DLMA` instead of `data_root`, etc.
* allows `data_root` to be a list.

By doing these things, both `EveryDreamBatch` and DLMA can be free from
data resolution logic. It also reduces the number of arguments which
need to be passed down to EDB and DLMA.
2023-01-29 17:08:54 -08:00
Victor Hall 18d1da0459 bug fix and multiplytxt fraction stuff 2023-01-22 18:59:59 -05:00
Jan Gerritsen 711e263e24 Implemented system to train on a subset of the dataset, favouring higher rated images 2023-01-14 16:18:22 +01:00
Jan Gerritsen 3d2709ace9 Implemented loading captions from yaml file 2023-01-09 21:53:07 +01:00
Jan Gerritsen a3618409bc Support more control regarding caption tag shuffeling using yaml files 2023-01-09 21:53:07 +01:00
Victor Hall 98f9a7302d shuffle tags arg 2023-01-06 19:12:52 -05:00
Victor Hall 65c5fd5ccb json driven args, update run batch compensation 2023-01-03 14:27:26 -05:00
Victor Hall 07125135a8 safer writing of batch schedule 2023-01-02 18:02:35 -05:00
Victor Hall b316684bdb bunch of updates, grad ckpting, no drop bucket, shuffle every epoch 2023-01-01 10:45:18 -05:00
Victor Hall 4c53f2d55c various tweaks and bugfixes over holidays 2022-12-27 14:25:32 -05:00
Victor Hall 8904724135 fix some quality issues 2022-12-20 03:30:42 -05:00
Victor Hall ac6391ef87 save ever n epochs, update docs 2022-12-18 21:16:14 -05:00
Victor Hall 179fd5395b hey look ed2 2022-12-17 22:32:48 -05:00