Skip to content

Releases: BradleyEdelman/EdgeTrain

EdgeTrain v0.2.0: Scoring and prioritization for dynamic parameter updates

11 Feb 16:16

Choose a tag to compare

This version introduces a refined adaptive training strategy with a constant pruning ratio. Key updates:

  • Score Calculation: This version now computes an accuracy score and a memory score based on resource usage and model performance.
  • Parameter Prioritization: Accuracy and memory scores are weighted according to default or user-defined priority weighting schemes to idenfity a priority order for parameter adjustment. Only the top priority paramater is adjusted in each epoch.
    • Batch size priority is weighted by memory usage.
    • Learning rate priority is inversely weighted by accuracy improvement (i.e. increases if accuracy stagnates).
  • Fixed Pruning Ratio: Pruning is constant and is stripped at the end.
  • Code Quality Improvements: Added pre-commit hooks and CI linting for consistency.

Bug Fix: Circular Import Issue

10 Jan 16:00

Choose a tag to compare

Pre-release

Fixed a circular import issue in the package to allow for proper imports at package level rather than module level.

EdgeTrain v0.1.0-alpha: Initial Release

10 Jan 16:12

Choose a tag to compare

Pre-release

EdgeTrain is a utility package designed for machine learning training with limited resources. This initial alpha release includes functionality to dynamically adjust batch sizes and learning rates based on available CPU/GPU resources, helping optimize training processes on edge devices.