Skip to content

Version 3

Compare
Choose a tag to compare
@muditbhargava66 muditbhargava66 released this 25 Apr 08:46
· 7 commits to main since this release

Updates in Version 3.0.0

  • Enhanced cross-platform compatibility: The codebase now works seamlessly on macOS, Windows, and Linux
  • Improved device selection logic: Automatically detects and utilizes the available hardware (MPS, CUDA, or CPU) for training
  • Updated dependencies: Added torchvision, torchaudio, matplotlib, and scipy as dependencies in requirements.txt and pyproject.toml
  • Improved visualization: Enhanced visualize.py with better plot layout and cross-platform file paths
  • Code cleanup and refactoring: Improved code structure and readability
  • Added mathematical analysis: Introduced mathematical_analysis.py to analyze the effect of DropGrad on various optimizers
  • Added benchmark visualizations: Introduced benchmark_visualizations.py to compare the behavior of DropGrad across optimizers and benchmarks