Minor:
- Disable mixed precision calls
- Move data IO management to biapy/data and create
imread
/imwrite
functions to avoid usingskimage.io
as they are deprecating these functions - Change
timm.optim
call to avoid warning of future deprecation - BioImage Model Zoo related (BMZ) changes:
- Add function to autogenerate a
documentation.md
- Move BMZ related functions to
bmz_utils.py
- Add test_model at the end of BMZ model creation with a larger tolerance than the default so the differences due to casting are allowed
- Create cover extracting a patch containing mask information
- Add argument to provide the dataset id when exporting a BMZ model
- Ensure
weights_only=True
during checkpoint loading stage when building a BMZ model
- Add function to autogenerate a
Bugs fixed:
- Checkpoint load on DDP right after training fixed
- Update installation restricting torchmetric version to avoid an issue in classification workflow
Full Changelog: v3.5.9...v3.5.10