Recent Releases of epochlib
epochlib - v4.1.0
Improvements
General
- Make numpy a required dependency (not optional)
CI/CD
- Configure pre-commit.ci to cache environments
- Dependabot merges to version branch workflow
Torch Trainer
- Unprivatise functions
- Support for mixed precision models
- Updated logging messages
- Support for custom dataset input
Data
- New dataset that can take training pipeline
- New data class based on enum retrieval
Cacher
- Reduced complexity
- Paths are now Path instead of str
- Python
Published by schobbejak over 1 year ago
epochlib - v4.0.0
Improvements
General
- Shorten import path lengths
- Add default logger
- Setup dependabot to manage dependencies
- Deploy documentation on GitHub Pages
CI/CD
- Only run tests on python version 3.10 every PR and on every version for release PRs
Documentation
- Add optional dependencies description to README.md
TIMM
- Refactor class to include
Torch Trainer
- Add ONNX/Openvino compile option to trainer
Augmentations
- Add BirdCLEF audio augmentations
Bug Fixes
General
- Fix PyPi logo
- Python
Published by schobbejak over 1 year ago
epochlib - v0.3.7
Improvements
General
- Allow project to be managed with Rye
- Update pre-commit configuration
- Add CITATION.cff file to allow project to be cited
Torch trainer
- Option for prefix and suffix for logging in torch trainer
Bug Fixes
Torch Trainer
- Make patience have default value -1 in trainer
- to_predict and tm folder are not considered in torch trainer hash
- Python
Published by schobbejak over 1 year ago
epochlib - v0.3.4
Improvements
Torch Trainer
- Implement Checkpoints (
checkpointing_enabled,checkpointing_keep_every,checkpointing_resume_if_exists). After each epoch, the model is saved to disk. By default only the current checkpoint is saved on disk, while old ones are deleted. This can also be changed. - A custom
collate_fnfor theDataLoaderclass is can now be passed intoTorchTrainer. This is useful for e.g. GNN's
Models/timm Improvements
- It is now possible to specify whether to use the pre-trained model or not.
PyTest
- Refactored Tests to use fixtures
- Resolve most warnings
Bug Fixes
Torch Trainer
- Now raises an error if
model_nameis not specified. - Now check if the specified
trained_models_directoryexists, and creates it. - Fix a bug in trainoneepoch: Copied
xinto the `ybatch, instead ofy`.
- Python
Published by justanotherariel over 1 year ago
epochlib - v0.3.3
Improvements
Torch Trainer
- Fix bug in scheduler step
- Add ability to specify dataloader arguments
- Add ability to specify type of the x and y tensors
Logging
- When cache is stored a message is logged to terminal, fixing issue with large datasets taking a long time to store with no indicator what is happening within the pipeline
- Python
Published by schobbejak almost 2 years ago
epochlib - v0.3
- Timm wrapper for models
- Updates to torch_trainer: Early stopping, save to external, fold ensemble support, prediction type
- Augmentations
- Update cache_args
- Fix optional imports not being fully optional
- Improvements to general readability
- Python
Published by schobbejak almost 2 years ago