southern-cross-ai / BabyJoey

Small 115 million parameter model - .5GB
Apache License 2.0
4 stars 9 forks source link

Refactor training logic setup Description: #32

Open matthewaltenburg opened 1 month ago

matthewaltenburg commented 1 month ago

Aim to simplify the setup of training logic by potentially integrating configurations directly into the BabyJoeyUnit constructor.