fork-2883-ditwoo/fix/RandomJPEG-device-handling - Branch - CodSpeedHQ/kornia

Fix: `RandomJPEG` bug with different devices(#24)

Comparing

fork-2883-ditwoo/fix/RandomJPEG-device-handling

(

6e631a2

) with

main

(

a5cdba9

)

test_aug_2d_linear_corner_illumination[1-1-128-128-cpu-float32-inductor]

benchmarks/augmentation/2d_intensity_test.py::test_aug_2d_linear_corner_illumination[1-1-128-128-cpu-float32-inductor]

4.2 s

test_aug_2d_solarize[1-1-128-128-cpu-float32-inductor]

benchmarks/augmentation/2d_intensity_test.py::test_aug_2d_solarize[1-1-128-128-cpu-float32-inductor]

2.6 s

test_aug_2d_thin_plate_spline[1-1-128-128-cpu-float32-inductor]

benchmarks/augmentation/2d_geometric_test.py::test_aug_2d_thin_plate_spline[1-1-128-128-cpu-float32-inductor]

7.2 s

test_aug_2d_elastic_transform[1-1-128-128-cpu-float32-inductor]

benchmarks/augmentation/2d_geometric_test.py::test_aug_2d_elastic_transform[1-1-128-128-cpu-float32-inductor]

8.1 s

Click on a commit to change the comparison range

base

main

a5cdba9

fix: move jpeg_quality tensor to device to handle different devices

6e631a2

6 months ago

by ditwoo