Skip to content

Version 0.3.5

Latest
Compare
Choose a tag to compare
@muditbhargava66 muditbhargava66 released this 27 Apr 05:38

What's New in Version 0.3.5?

  • Added support for the Lion optimizer in the ViT experiments
  • Implemented gradient clipping to prevent gradient explosion and improve training stability
  • Enhanced data augmentation techniques for better model generalization
  • Improved error handling and user interruption handling during training
  • Updated test suite to cover various aspects of DropGrad, including initialization, optimization step, drop rate scheduling, and saving of loss values
  • Code refactoring and documentation enhancements for better readability and maintainability