search

decile-team / cords

Reduce end to end training time from days to hours (or hours to minutes), and energy requirements/costs by an order of magnitude using coresets and data selection.
https://cords.readthedocs.io/en/latest/
MIT License
316 stars 53 forks source link

Evaluation on ImageNet #35

Closed Hayoung93 closed 2 years ago

Hayoung93 commented 2 years ago

Hello, thanks for a very interesting and useful project.

Could you mind providing an evaluation method for ImageNet? I tried to, adding loader for ImageNet to custom_dataset.py, but failed due to a GPU memory issue during subset selection.

Many thanks!

krishnatejakk commented 2 years ago

@Hayoung93 Sorry for the delay. I will add the notebook for ImageNet evaluation shortly in examples.