LUNA / train.py

Commit History

Upload train.py with huggingface_hub
28e9a14
verified

BHARGAV REDDY commited on

Bulletproof probe: simulate grad_accum, expandable_segments, 70% safety
4d7a400

ASTERIZER commited on

Fix: probe with return_logits=False (was wasting 3GB on logits tensor), show peak VRAM
4d6a3fe

ASTERIZER commited on

Fix probe: use actual model (no duplicate), add GPU cleanup at start
78a081e

ASTERIZER commited on

Maximize GPU usage: probe up to 4096 batch, 92% safety, only catch OOM
80bdec5

ASTERIZER commited on

Disable torch.compile entirely - causes OOM/crash at 100M scale
649da09

ASTERIZER commited on

Fix: use compile mode=default to avoid CUDA graph + tied embedding crash
0a6cf2e

ASTERIZER commited on

Fix torch.compile CUDA graph training on 5090
4bf87a0

ASTERIZER commited on

LUNA 100M: cloud-ready training pipeline
ad68b7f

ASTERIZER commited on