wavedl 1.6.0__py3-none-any.whl → 1.6.1__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (37) hide show
  1. wavedl/__init__.py +1 -1
  2. wavedl/hpo.py +451 -451
  3. wavedl/models/__init__.py +28 -0
  4. wavedl/models/{_timm_utils.py → _pretrained_utils.py} +128 -0
  5. wavedl/models/base.py +48 -0
  6. wavedl/models/caformer.py +1 -1
  7. wavedl/models/cnn.py +2 -27
  8. wavedl/models/convnext.py +5 -18
  9. wavedl/models/convnext_v2.py +6 -22
  10. wavedl/models/densenet.py +5 -18
  11. wavedl/models/efficientnetv2.py +315 -315
  12. wavedl/models/efficientvit.py +398 -0
  13. wavedl/models/fastvit.py +6 -39
  14. wavedl/models/mamba.py +44 -24
  15. wavedl/models/maxvit.py +51 -48
  16. wavedl/models/mobilenetv3.py +295 -295
  17. wavedl/models/regnet.py +406 -406
  18. wavedl/models/resnet.py +14 -56
  19. wavedl/models/resnet3d.py +258 -258
  20. wavedl/models/swin.py +443 -443
  21. wavedl/models/tcn.py +393 -409
  22. wavedl/models/unet.py +1 -5
  23. wavedl/models/unireplknet.py +491 -0
  24. wavedl/models/vit.py +3 -3
  25. wavedl/train.py +1430 -1430
  26. wavedl/utils/config.py +367 -367
  27. wavedl/utils/cross_validation.py +530 -530
  28. wavedl/utils/losses.py +216 -216
  29. wavedl/utils/optimizers.py +216 -216
  30. wavedl/utils/schedulers.py +251 -251
  31. {wavedl-1.6.0.dist-info → wavedl-1.6.1.dist-info}/METADATA +93 -53
  32. wavedl-1.6.1.dist-info/RECORD +46 -0
  33. wavedl-1.6.0.dist-info/RECORD +0 -44
  34. {wavedl-1.6.0.dist-info → wavedl-1.6.1.dist-info}/LICENSE +0 -0
  35. {wavedl-1.6.0.dist-info → wavedl-1.6.1.dist-info}/WHEEL +0 -0
  36. {wavedl-1.6.0.dist-info → wavedl-1.6.1.dist-info}/entry_points.txt +0 -0
  37. {wavedl-1.6.0.dist-info → wavedl-1.6.1.dist-info}/top_level.txt +0 -0
wavedl/__init__.py CHANGED
@@ -18,7 +18,7 @@ For inference:
18
18
  # or: python -m wavedl.test --checkpoint best_checkpoint --data_path test.npz
19
19
  """
20
20
 
21
- __version__ = "1.6.0"
21
+ __version__ = "1.6.1"
22
22
  __author__ = "Ductho Le"
23
23
  __email__ = "ductho.le@outlook.com"
24
24