main.py: --resume-phase, --checkpoint-dir, --sinkhorn-batch flags a365009 verified rogermt commited on 27 minutes ago
Add checkpointing, resume, CIFAR OOM fix, --sinkhorn-batch flag 8b62ba9 verified rogermt commited on 38 minutes ago
Update SKILL.md with DataLoader batch size mismatch bug and shared state pitfall d5802bb verified rogermt commited on about 2 hours ago
--train-iters now also overrides time_predictor.num_iterations 7921a70 verified rogermt commited on about 2 hours ago
Fix DataLoader batch size mismatch across training phases + --train-iters now overrides all phases 9e3fccc verified rogermt commited on about 3 hours ago
Add pool.finalize() call after building trajectory pool for O(1) sampling 376238e verified rogermt commited on about 4 hours ago
Fix geomloss tensor shape bug for images + optimize pool sampling 3e32ac2 verified rogermt commited on about 4 hours ago
Add SKILL.md — paper reproduction skill with lessons from NSGF++ implementation 88f3058 verified rogermt commited on about 4 hours ago