Skip to content

Release 2.0.2 - Validator Logging and VRAM Consumption
Compare
Choose a tag to compare
@dylanuys dylanuys released this 11 Dec 19:50
· 1 commit to main since this release
v2.0.2
b76f5de
This commit was created on GitHub.com and signed with GitHub’s verified signature.
GPG key ID: B5690EEEBB952194
Learn about vigilant mode.
  • Fix for info logs to show by default for validator process
  • Ensure that processes that do not require GPU (cache updater, validator) do not allocate any VRAM for cuda/torch configurations.
    • This reduces overall VRAM usage by ~4-5GB
  • try/except for wandb challenge data population to avoid errors when corrupt images are sampled (due to a failed extraction from parquet)
Assets 2
Loading