Repository navigation
v0.2.0
pixtreme-train v0.2.0
New training components in pixtreme.train.common:
write_textandstaging_directoryreplace a file or a directory only after the new content is fully written.training_signalsturns SIGINT and SIGTERM into a stop request for a training loop, withdefer()for sections that must finish.worker_signalsandstop_loader_workershandle signals in DataLoader worker processes and stop and reap the workers of an iterator without waiting for prefetched batches.rng_stateandrestore_rng_statesave and restore the Python, NumPy, torch CPU and CUDA, and namedtorch.Generatorrandom states.warmup_cosine_factor,update_ema,check_finite_loss,check_finite_gradients,release_gpu_memoryandEndlessSamplerprovide small training parts.
Importing pixtreme.train.common still does not load torch; each torch-based name loads its module on first access.
The wheel attached here is the reference copy for the PyPI distribution of pixtreme-train. Verify it with sha256sum -c SHA256SUMS.