ARC-AGI / trm_solver /trm_nano /train_task.py

Commit History

Fix: force unbuffered stdout for Kaggle notebook output + use print() with flush"
0e4cf22
verified

rogermt commited on

Add --log_batch_every: prints loss every N batches within an epoch"
e997a38
verified

rogermt commited on

Add --num_gpus: splits tasks across GPUs in parallel processes"
626c4f7
verified

rogermt commited on

Add --log_every arg to control epoch print frequency (default 100)
e0b6ea4
verified

rogermt commited on

Replace train_task.py with full batch runner: loops all 400 tasks, skips solved, logs everything, uses manual ONNX builder
28954c6
verified

rogermt commited on

Add TRM-Nano per-task training script with augmentation and ONNX export
7dd4c24
verified

rogermt commited on