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 |