3.3.1

版本发布时间: 2023-10-26 23:17:53

Deci-AI/super-gradients最新发布版本:3.7.1(2024-04-09 00:50:10)

Super Gradients 3.3.1

Improvements

Add class_names to model.predict for detection by @Louis-Dupont in https://github.com/Deci-AI/super-gradients/pull/1529
Data-Gradients dependency bumped to 0.2.2 by @Louis-Dupont in https://github.com/Deci-AI/super-gradients/pull/1541
Check class_id validity in DetectionDataset by @Louis-Dupont in https://github.com/Deci-AI/super-gradients/pull/1536
Fixing typos and adding extra info to CONTRIBUTING by @hakuryuu96 in https://github.com/Deci-AI/super-gradients/pull/1546

ListConfig/DictConfig types do not leak into checkpoint state dictionary anymore by @BloodAxe in https://github.com/Deci-AI/super-gradients/pull/1534
Migrate usage of np.bool -> bool which is not supported in latest np versions by @Louis-Dupont in https://github.com/Deci-AI/super-gradients/pull/1558
Ensure that checkpoint_num_classes is propagated from YAML files to models.get by @BloodAxe in https://github.com/Deci-AI/super-gradients/pull/1533
Fixed detection models export to ONNX bug for batch size > 1 @BloodAxe in https://github.com/Deci-AI/super-gradients/pull/1530
Fix CSPDarknet53.foward by @Louis-Dupont in https://github.com/Deci-AI/super-gradients/pull/1564
Fixed incorrect automatic variable used by @BloodAxe in https://github.com/Deci-AI/super-gradients/pull/1565
Fix typo in class documentation by @aler9 in https://github.com/Deci-AI/super-gradients/pull/1548
Trainer does not crash anymore when using CPU and Automated Mixed Precision is enabled in training params @Louis-Dupont in https://github.com/Deci-AI/super-gradients/pull/1567
Fixed issue with torch 1.12 where _scale_fn_ref is missing in CyclicLR by @BloodAxe in https://github.com/Deci-AI/super-gradients/pull/1575
Fixed issue with torch 1.12 issue with arange not supporting fp16 for CPU device. by @BloodAxe in https://github.com/Deci-AI/super-gradients/pull/1574

Full Changelog: https://github.com/Deci-AI/super-gradients/compare/3.3.0...3.3.1