Pedro Rodriguez ab399e981d Several changes to enable entropy model training/eval ... Summary: - Make arrow iterator able to read from jsonl files, the entropies are omitted in this case - Make the data/checkpoint code fsspec compatible - Fix issues with all reduce with non-bf16 in dist_sum and norm computation. - Minimal fixes to get eval to run, it is slow currently - Add bpb numbers during training Test Plan: Run ``` torchrun --nproc-per-node 8 -m bytelatent.train config=internal/configs/entropy_model.yaml eval=null max_steps=10100 ```		2025-02-04 18:05:16 +00:00
..
iterators	Several changes to enable entropy model training/eval	2025-02-04 18:05:16 +00:00
__init__.py	Initial commit	2024-12-12 15:32:30 -08:00
data_types.py	This includes fixes that make checkpointing and reloading work correctly. (#35)	2025-01-27 16:56:42 -08:00
file_util.py	Update file check script to check sizes (#32)	2025-01-22 13:06:46 -08:00
ngram_processor.py	Initial commit	2024-12-12 15:32:30 -08:00
patcher.py	Initial codes and scripts for training entropy model (#34)	2025-01-27 09:46:44 -08:00