mrq 74df2f5332 split sampler dict by global_rank, also handle splitting dataset paths by global_rank if sampler_type == path (because I do not trust DistributedSampler) (need to test)		2024-06-01 09:29:49 -05:00
..
emb	DAC just doesn't work well enough......	2024-05-25 11:07:52 -05:00
engines
ext
models	added model config option to set KV head count for MQA/GQA instead of MHA for llama-based models (i think its very negligible both ways on such a small model size)	2024-05-31 19:32:37 -05:00
utils	some cleanup	2024-05-25 17:46:52 -05:00
__init__.py
__main__.py
config.py	actually don't default to compute split losses, test bitnet model doesn't seem to be doing things right (despite debug printouts showing theyre roughly the same logit/loss sequences, could just be bitnet linears being not up to par on actual models)	2024-06-01 09:12:51 -05:00
data.py	split sampler dict by global_rank, also handle splitting dataset paths by global_rank if sampler_type == path (because I do not trust DistributedSampler) (need to test)	2024-06-01 09:29:49 -05:00
export.py
inference.py	DAC just doesn't work well enough......	2024-05-25 11:07:52 -05:00
plot.py
samplers.py
train.py	nevermind it breaks training	2024-05-25 18:03:43 -05:00
webui.py