search

vturrisi / solo-learn

solo-learn: a library of self-supervised methods for visual representation learning powered by Pytorch Lightning
MIT License
1.39k stars 181 forks source link

add `dali_device` option to the Resize op #311

Closed ojss closed 1 year ago

ojss commented 1 year ago

This will support running the Resize op on the GPU.

Using the vanilla Resize op fails without the device key since it runs on the CPU by default. A small change in the code fixes this.

cc/ @anujinho

codecov[bot] commented 1 year ago

Codecov Report

Merging #311 (b10a6ec) into main (1884bd7) will not change coverage. The diff coverage is n/a.

Additional details and impacted files | Flag | Coverage Δ | | *Carryforward flag | |---|---|---|---| | cpu | `81.42% <ø> (ø)` | | | | dali | `39.38% <ø> (ø)` | | Carriedforward from [1884bd7](https://codecov.io/gh/vturrisi/solo-learn/commit/1884bd7d1f611ae79e356c96c6f43addcb18c4cd?utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=Victor+Turrisi) | *This pull request uses carry forward flags. [Click here](https://docs.codecov.io/docs/carryforward-flags?utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=Victor+Turrisi) to find out more. | [Impacted Files](https://codecov.io/gh/vturrisi/solo-learn/pull/311?src=pr&el=tree&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=Victor+Turrisi) | Coverage Δ | | |---|---|---| | [solo/data/dali\_dataloader.py](https://codecov.io/gh/vturrisi/solo-learn/pull/311/diff?src=pr&el=tree&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=Victor+Turrisi#diff-c29sby9kYXRhL2RhbGlfZGF0YWxvYWRlci5weQ==) | `91.84% <ø> (ø)` | |