tinyvla / tinyvla_b200 /scripts /train_fast.py

Commit History

CRITICAL fix: torch.compile'd expert saved as expert._orig_mod.* -> from_pretrained silently dropped it (random flow head at eval). Save the uncompiled module
f0fcec5
verified

AlexWortega commited on

train_fast: non-persistent hub workers (recycle each epoch to reset leaks)
d990e88
verified

AlexWortega commited on

train_fast: v4 all-channels via hub stream (support passthrough, morph_text tokenization, guarded shard morph block)
d9b76b2
verified

AlexWortega commited on

fix DDP hang: spawn dataloader workers (fork-after-CUDA/NCCL starved torchrun heartbeat), picklable Collate, fewer workers per rank
4ac1f5f
verified

AlexWortega commited on

fix: yaml 1.1 parses '2250.0e12' as string -> float() cast in MFU calc (crashed first log step)
fd625f9
verified

AlexWortega commited on

train_fast: torchrun DDP (per-rank streaming, rank0 logging/saves, global throughput)
0febc93
verified

AlexWortega commited on

tinyvla_b200: B200 streaming training on nvidia/physical-ai (745 samples/s e2e, 7x ViT fast path, hub episode streaming, full instructions)
51ce5b3
verified

AlexWortega commited on