Commit Graph

8 Commits

Author SHA1 Message Date
Jan Gerritsen f47ceadcc7 Implemented an optimization for the shuffling if all tags have the same weight and added documentation. 2023-01-09 21:53:46 +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 b316684bdb bunch of updates, grad ckpting, no drop bucket, shuffle every epoch 2023-01-01 10:45:18 -05:00
Victor Hall 051116a7d9 add gpu id support 2022-12-29 21:11:06 -05:00
Victor Hall 4c53f2d55c various tweaks and bugfixes over holidays 2022-12-27 14:25:32 -05:00
Victor Hall aff2377209 docs, update setup 2022-12-18 13:03:44 -05:00
Victor Hall 179fd5395b hey look ed2 2022-12-17 22:32:48 -05:00