fork-3005-hit2sjtu/dev - Branch - CodSpeedHQ/kornia

Fix type mis-match during mixed-precision training with extract_patches_from_pyramid

Comparing

fork-3005-hit2sjtu/dev

(

4058274

) with

main

(

d645a84

)

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

benchmarks/augmentation/2d_geometric_test.py

+1%

7.8 s7.8 s

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

benchmarks/augmentation/2d_geometric_test.py

8.5 s8.5 s

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

benchmarks/augmentation/2d_intensity_test.py

3 s3 s

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

benchmarks/augmentation/2d_intensity_test.py

-1%

4.6 s4.6 s

Click on a commit to change the comparison range

Base

main

d645a84

-0.1%

Fix type mis-match during mixed-precision training. Out and patches can be (Float Half), thus cause issue. Signed-off-by: Qiang Wang <wq@sightp.com>

4058274

1 year ago

by hit2sjtu