search

southern-cross-ai / BabyJoey

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

Add SPLIT_RATIO into config.py #20

Closed iamyifan closed 1 month ago

iamyifan commented 1 month ago

Change the former hard-coded split_ratio=0.2 in BabyJoeyDataset to a hyperparameter in config.py

iamyifan commented 1 month ago

Tested and merged into origin/main