#257 allow using an external Optimizer (not initialized outside)

Merged

GitHub User merged 1 commits into Deci-AI:master from deci-ai:feature/SG-184_external_optimizer

  
1

2

3

4

5

6

7

8

9

10

11

12

            from super_gradients.training.datasets import CoCoDetectionDatasetInterface
from super_gradients.training.utils.detection_utils import AnchorGenerator
dataset_params = {"batch_size": 1,
                  "test_batch_size": 1,
                  "dataset_dir": "/data/coco/",
                  "image_size": 640,
                  "train_sample_loading_method": 'rectangular'
                  }
coco_dataset_interface = CoCoDetectionDatasetInterface(dataset_params=dataset_params, cache_labels=True)
anchors = AnchorGenerator(coco_dataset_interface.trainset, 9)

Tip!

Press p or to see the previous file or, n or to see the next file

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

#257 allow using an external Optimizer (not initialized outside)

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