Phil Wang
|
0f4edff214
|
derived value for image preprocessing belongs to the data config class
|
2022-05-22 18:42:40 -07:00 |
|
Phil Wang
|
501a8c7c46
|
small cleanup
|
2022-05-22 15:39:38 -07:00 |
|
Phil Wang
|
c12e067178
|
let the pydantic config base model take care of loading configuration from json path
|
2022-05-22 14:47:23 -07:00 |
|
Phil Wang
|
c6629c431a
|
make training splits into its own pydantic base model, validate it sums to 1, make decoder script cleaner
|
2022-05-22 14:43:22 -07:00 |
|
Phil Wang
|
a1ef023193
|
use pydantic to manage decoder training configs + defaults and refactor training script
|
2022-05-22 14:27:40 -07:00 |
|
Phil Wang
|
b432df2f7b
|
final cleanup to decoder script
|
2022-05-21 10:42:16 -07:00 |
|
Phil Wang
|
8b0d459b25
|
move config parsing logic to own file, consider whether to find an off-the-shelf solution at future date
|
2022-05-21 10:30:10 -07:00 |
|
Phil Wang
|
0064661729
|
small cleanup of decoder train script
|
2022-05-21 10:17:13 -07:00 |
|
Phil Wang
|
80497e9839
|
accept unets as list for decoder
|
2022-05-20 20:31:26 -07:00 |
|
Phil Wang
|
8997f178d6
|
small cleanup with timer
|
2022-05-20 20:05:01 -07:00 |
|
Aidan Dempster
|
022c94e443
|
Added single GPU training script for decoder (#108)
Added config files for training
Changed example image generation to be more efficient
Added configuration description to README
Removed unused import
|
2022-05-20 19:46:19 -07:00 |
|