3.3.1

Super Gradients 3.3.1

Improvements

• Add class_names to model.predict for detection by @Louis-Dupont in #1529
• Data-Gradients dependency bumped to 0.2.2 by @Louis-Dupont in #1541
• Check class_id validity in DetectionDataset by @Louis-Dupont in #1536
• Fixing typos and adding extra info to CONTRIBUTING by @hakuryuu96 in #1546
• Progress bar during training now respects max_train_batches/max_valid_batches by @hakuryuu96 #1554

Bugfixes

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

New Contributors

• @hakuryuu96 made their first contribution in #1546
• @aler9 made their first contribution in #1548

Full Changelog: 3.3.0...3.3.1