Skip to content

Release v0.3.1
Compare
Choose a tag to compare
@mathpluscode mathpluscode released this 25 Nov 02:55
· 10 commits to main since this release
v0.3.1
5349fdb
This commit was created on GitHub.com and signed with GitHub’s verified signature. The key has expired.
GPG key ID: 4AEE18F83AFDEB23
Expired
Learn about vigilant mode.

Added

  • Added example notebooks for inference on a single image without TFDS.
  • Added integration tests for training, validation, and testing.

Changed

  • ⚠️ Upgrade to JAX to 0.4.20.
  • ⚠️ Removed Haiku-specific modification to convolutional layers. This may impact model
    performance.
  • Refactored config.
    • Added patch_size and scale_factor to data config.
    • Moved loss config from the main config to task config.
  • Refactored code, including defining imgx/task submodule.
Assets 2
Loading