google-deepmind / google-deepmind/dm_pix
Gaussian blur cpu performance
- Langage dominant
- Python
- Étoiles
- 448
- Forks
- 31
- Merge moyen
- 1 j 55 min
- PR mergées (30 j)
- 5
Description
I have been doing some experiments with PIX since it allows computing image augmentations in the GPU in contrast to torchvision which computes in the CPU and requires multiple workers to avoid bottlenecks. When performing some very simple `timeit` examples I observed a very high time when performing a gaussian blur in the CPU. I created a simple [Colab notebook](https://colab.research.google.com/drive/1UgqZVW2I09pPia2EZKhsZTnvwvucRu3Z?usp=sharing ) to demonstrate these experiments. I even tested transferring the image to CPU before performing the blur but it doesn't seem to make any difference. I was wondering if this is intended and I should not rely on CPU computations at all or if something is yet to be optimized for CPU computation.
Guide de contribution
Ouvrir le guide de contribution
Piste de recherche
Commencez par le notebook Colab lié et ses expériences timeit pour Gaussian blur, puis suivez le chemin d’exécution CPU utilisé par l’opération de flou dans dm_pix. Comparez les temps d’exécution CPU et GPU et déterminez si le coût observé est attendu ou s’il indique une possibilité d’optimisation. La tâche est terminée lorsque la cause est documentée et qu’un objectif ou une modification concrète de performance a été convenu.
Rédigé par le modèle d'indexation à partir du texte de l'issue.
Évaluation
- Stack technique
- python
- Domaine
- computer-vision, machine-learning, performance
- Type d'issue
- Bug
- Difficulté
- 4/5
- Temps estimé
- 3-5 jours
- Activité
- À l'abandon
- Clarté
- À clarifier
- Accessibilité débutants
- 28/100