mrq 4800e7179a remove nan checks because it causes problems in distributed training because I'm not syncing between GPUs (and nan losses gets ignored anyways with loss scaling)		2024-12-15 09:42:54 -06:00
..
emb
engines	remove nan checks because it causes problems in distributed training because I'm not syncing between GPUs (and nan losses gets ignored anyways with loss scaling)	2024-12-15 09:42:54 -06:00
models	APOLLO cringe (doesn't want to work with deepspeed)	2024-12-12 00:31:58 -06:00
utils	ugh	2024-12-14 22:43:51 -06:00
__init__.py
__main__.py
config.py	APOLLO tweaks to make it work with deepspeed	2024-12-13 23:03:52 -06:00
data.py	APOLLO cringe (doesn't want to work with deepspeed)	2024-12-12 00:31:58 -06:00
demo.py	more fixes for local engine backend	2024-12-12 14:38:42 -06:00
export.py
inference.py
metrics.py
plot.py
samplers.py
train.py	remove nan checks because it causes problems in distributed training because I'm not syncing between GPUs (and nan losses gets ignored anyways with loss scaling)	2024-12-15 09:42:54 -06:00
webui.py