rogermt
/

nsgf-plusplus

Model card Files Files and versions

nsgf-plusplus / trainer.py

Commit History

Add checkpointing, resume, CIFAR OOM fix, --sinkhorn-batch flag

8b62ba9
verified

rogermt commited on 13 days ago

Add pool.finalize() call after building trajectory pool for O(1) sampling

376238e
verified

rogermt commited on 13 days ago

Upload trainer.py

81d2fba
verified

rogermt commited on 15 days ago