Deci-AI/super-gradients

Louis-Dupont b8848d06d0

Make classification models inherit from BaseClassifier (#1314)

2 years ago

Commits

__init__.py

b8848d06d0

Make classification models inherit from BaseClassifier (#1314)

2 years ago

all_architectures_test.py

71cc471b86

Feature/sg 691 register dynamically metrics and datasets (#744)

2 years ago

average_meter_test.py

c4b4e37e27

apply black on tests (#633)

2 years ago

bbox_formats_test.py

3f443b2b9f

fix: Prevent a wrong warning (#779)

2 years ago

cityscapes_dataset_test.py

b4608f618a

Cityscapes AutoLabelling dataset (#1000)

2 years ago

coco_segmentation_dataset_test.py

c4b4e37e27

apply black on tests (#633)

2 years ago

config_inspector_test.py

3fa5936e79

Fix AccessCounterMixin (#705)

2 years ago

conv_bn_relu_test.py

c4b4e37e27

apply black on tests (#633)

2 years ago

crash_tips_test.py

79418efd4f

Interpollation crash tip (#616)

2 years ago

dataloader_factory_test.py

7d2d658081

Remove InfiniteSampler (#883)

2 years ago

dataset_statistics_test.py

07ec5e5b63

Feature/sg 000 introduce class agnoistic nms option (#1232)

2 years ago

ddr_loss_test.py

c4b4e37e27

apply black on tests (#633)

2 years ago

dekr_loss_test.py

7bede95d2a

Feature/sg 572 pose estimation loss function (#709)

2 years ago

detection_caching.py

c4b4e37e27

apply black on tests (#633)

2 years ago

detection_dataset_test.py

f4a0d9e5d1

moved exceptions from training (#1260)

2 years ago

detection_output_adapter_test.py

9eb59e128c

Added tutorial on DetectionOutputAdapter (#817)

2 years ago

detection_sub_classing_test.py

f4a0d9e5d1

moved exceptions from training (#1260)

2 years ago

detection_sub_sampling_test.py

c4b4e37e27

apply black on tests (#633)

2 years ago

detection_targets_format_transform_test.py

75b6a4fd72

Hotfix/alg 1470 drop boxes padding (#1107)

2 years ago

detection_utils_test.py

07ec5e5b63

Feature/sg 000 introduce class agnoistic nms option (#1232)

2 years ago

dice_loss_test.py

b8915f15a1

black (#576)

2 years ago

double_training_test.py

1f6a183671

Feature/sg 493 modelnames instead of strings (#614)

2 years ago

early_stop_test.py

c4b4e37e27

apply black on tests (#633)

2 years ago

export_coreml_test.py

a9cee54569

Added CoreML export notebook & method for Apple devices. (#1068)

2 years ago

export_onnx_test.py

a9cee54569

Added CoreML export notebook & method for Apple devices. (#1068)

2 years ago

export_utils_test.py

71cc471b86

Feature/sg 691 register dynamically metrics and datasets (#744)

2 years ago

factories_test.py

39b526aad5

Allow activation factory type to take resolved type as input (#797)

2 years ago

forward_pass_prep_fn_test.py

1f6a183671

Feature/sg 493 modelnames instead of strings (#614)

2 years ago

hydra_resolvers_test.py

788aade6aa

refacto environment package (#548)

2 years ago

initialize_with_dataloaders_test.py

41d455f917

Return Dict of loss/metrics instead of Tuple in the Trainer (#1218)

2 years ago

iou_loss_test.py

b8915f15a1

black (#576)

2 years ago

kd_ema_test.py

25463233a0

Getting rid of "module." heritage (#1184)

2 years ago

kd_trainer_test.py

25463233a0

Getting rid of "module." heritage (#1184)

2 years ago

load_checkpoint_test.py

bc82b294ad

Allow checkpoint_path to be an url (Required for transfer learning) (#1138)

2 years ago

load_ema_ckpt_test.py

c4b4e37e27

apply black on tests (#633)

2 years ago

local_ckpt_head_replacement_test.py

1f6a183671

Feature/sg 493 modelnames instead of strings (#614)

2 years ago

loss_loggings_test.py

1f6a183671

Feature/sg 493 modelnames instead of strings (#614)

2 years ago

lr_cooldown_test.py

b9fdd0e875

Feature/sg 850 support torch 2.0 (#944)

2 years ago

lr_warmup_test.py

b9fdd0e875

Feature/sg 850 support torch 2.0 (#944)

2 years ago

mask_loss_test.py

f5cfccc8c7

empty mask (#982)

2 years ago

max_batches_loop_break_test.py

d6f84a7030

Feature/sg 000 break inner train loop (#617)

2 years ago

multi_scaling_test.py

c4b4e37e27

apply black on tests (#633)

2 years ago

multiple_ignore_indices_segmentation_metrics_test.py

a2201ae735

Feature/sg 858 ignore multiple labels segmentation metrics support (#1177)

2 years ago

ohem_loss_test.py

c4b4e37e27

apply black on tests (#633)

2 years ago

optimizer_params_override_test.py

eb2ff83e34

Bug/sg 318 default optimizer params not taken with zero wd on bn and bias (#452)

2 years ago

phase_context_test.py

c4b4e37e27

apply black on tests (#633)

2 years ago

pose_estimation_dataset_test.py

a43cfcd700

predict() support for pose estimation models (#1142)

2 years ago

pose_estimation_metrics_test.py

69f08397bb

Feature/sg 636 pose estimation metrics (#717)

2 years ago

post_prediction_callback_test.py

d7452659e8

support single output in YoloX nms (#813)

2 years ago

ppyoloe_unit_test.py

12a4e53a96

PP-YoloE (#643)

2 years ago

preprocessing_unit_test.py

305a697bf8

Feature/sg 000 classwise ap (#1238)

2 years ago

pretrained_models_unit_test.py

1f6a183671

Feature/sg 493 modelnames instead of strings (#614)

2 years ago

quantization_utility_tests.py

537a0f0afe

Added qat_from_recipe script (#659)

2 years ago

random_erase_test.py

c4b4e37e27

apply black on tests (#633)

2 years ago

registry_test.py

71cc471b86

Feature/sg 691 register dynamically metrics and datasets (#744)

2 years ago

regnet_unit_test.py

c4b4e37e27

apply black on tests (#633)

2 years ago

replace_head_test.py

ab2e792bb8

Fix DEKR's replace_head & improve __repr__ for keypoints transforms (#1227)

2 years ago

repvgg_block_tests.py

71de0e3a46

Fix bug of not respecting dilation argument in RepVGGBlock (#555)

2 years ago

repvgg_unit_test.py

71cc471b86

Feature/sg 691 register dynamically metrics and datasets (#744)

2 years ago

resume_training_test.py

58c0915108

Bug/sg 399 external checkpoints fix (#468)

2 years ago

save_ckpt_test.py

1f6a183671

Feature/sg 493 modelnames instead of strings (#614)

2 years ago

segmentation_transforms_test.py

c4b4e37e27

apply black on tests (#633)

2 years ago

shelfnet_unit_test.py

c4b4e37e27

apply black on tests (#633)

2 years ago

strictload_enum_test.py

71cc471b86

Feature/sg 691 register dynamically metrics and datasets (#744)

2 years ago

test_auto_augment.py

c4b4e37e27

apply black on tests (#633)

2 years ago

test_deprecations.py

75b6a4fd72

Hotfix/alg 1470 drop boxes padding (#1107)

2 years ago

test_min_samples_single_node.py

cf365b36ff

Bug/sg 897 support for min samples with ddp (#1128)

2 years ago

test_predict.py

b8848d06d0

Make classification models inherit from BaseClassifier (#1314)

2 years ago

test_train_with_torch_scheduler.py

e9a99ac4ba

Feature/sg 946 simpify torch schedulers integration (#1230)

2 years ago

test_without_train_test.py

07ec5e5b63

Feature/sg 000 introduce class agnoistic nms option (#1232)

2 years ago

train_after_test_test.py

1da7133157

Adding test loaders (#1241)

2 years ago

train_logging_test.py

92de053378

Import AutoLoggerConfig and ConsoleSink separatly (#567)

2 years ago

train_with_intialized_param_args_test.py

1f6a183671

Feature/sg 493 modelnames instead of strings (#614)

2 years ago

train_with_precise_bn_test.py

c4b4e37e27

apply black on tests (#633)

2 years ago

training_params_factory_test.py

c4b4e37e27

apply black on tests (#633)

2 years ago

training_utils_test.py

49e9225bcd

Added TestTrainingUtils suite to unit tests that was missing for some reason (#621)

2 years ago

transforms_test.py

67970d476f

Add DetectionVerticalFlip (#1237)

2 years ago

update_param_groups_unit_test.py

c4b4e37e27

apply black on tests (#633)

2 years ago

vit_unit_test.py

1f6a183671

Feature/sg 493 modelnames instead of strings (#614)

2 years ago

yolo_nas_tests.py

e1db4d9949

Fix the bug with YOLONAS of not supporting overriding in_channels (#923)

2 years ago

yolox_unit_test.py

7907c486d1

Feature/sg 815 fix override dataset params (#1092)

2 years ago

zero_weight_decay_on_bias_bn_test.py

c4b4e37e27

apply black on tests (#633)

2 years ago

Comments

Use AWS S3 as storage!

Specify your S3 bucket

Access key (If needed)

Congratulations!

Use Google Cloud Storage!

Specify your Google Storage bucket

Service Account Key

Congratulations!

Use Azure Cloud Storage!

Specify your Azure Storage bucket

Access key (If needed)

Congratulations!

Use any S3 compatible storage!

Specify your S3 bucket

Access key (If needed)

Congratulations!

Deci-AI / super-gradients connected to https://github.com/Deci-AI/super-gradients.git

Comments

Use AWS S3 as storage!

Specify your S3 bucket

Access key (If needed)

Congratulations!

Use Google Cloud Storage!

Specify your Google Storage bucket

Service Account Key

Congratulations!

Use Azure Cloud Storage!

Specify your Azure Storage bucket

Access key (If needed)

Congratulations!

Use any S3 compatible storage!

Specify your S3 bucket

Access key (If needed)

Congratulations!

Deci-AI
/
super-gradients
connected to https://github.com/Deci-AI/super-gradients.git