search

ContinualAI / avalanche

Avalanche: an End-to-End Library for Continual Learning based on PyTorch.
http://avalanche.continualai.org
MIT License
1.71k stars 280 forks source link

Fixed detection api retrieval from dataset #1636

Open BlindedShooter opened 2 months ago

BlindedShooter commented 2 months ago

Added _FlatDataWithTransform to recursion condition of the function "get_detection_api_from_dataset".

Previously it recursed when the dataset is an instance of (Subset, AvalancheDataset, ConcatDataset),

but _FlatDataWithTransform can be given as "dataset" parameter in the recursion. (examples/detection_lvis.py)