Kevin Black
|
8779f62a1c
|
Adding checkpointing and resuming
|
2023-06-28 17:58:25 -07:00 |
|
Kevin Black
|
ad28862b48
|
Add reward to image caption
|
2023-06-28 10:42:47 -07:00 |
|
Kevin Black
|
28d2d8c40e
|
Minor changes; add train_timestep_fraction
|
2023-06-27 22:17:56 -07:00 |
|
Kevin Black
|
bae3f43f5f
|
Add aesthetic scorer reward function
|
2023-06-27 10:40:36 -07:00 |
|
Kevin Black
|
8cab96dea4
|
Minor changes, add assets
|
2023-06-27 10:20:03 -07:00 |
|
Kevin Black
|
4c5322ca85
|
Device specific seed
|
2023-06-26 22:35:24 -07:00 |
|
Kevin Black
|
5c16a90ceb
|
Move config out of module
|
2023-06-25 21:02:27 -07:00 |
|
Kevin Black
|
269615a35e
|
Working non-lora training; other changes
|
2023-06-25 11:28:42 -07:00 |
|
Kevin Black
|
c680890d5c
|
Working on DGX
|
2023-06-24 00:07:55 -07:00 |
|
Kevin Black
|
92fc030123
|
Continue implementation
|
2023-06-23 21:08:32 -07:00 |
|
Kevin Black
|
2fda3d4e78
|
Initial commit
|
2023-06-23 19:25:54 -07:00 |
|