Fix type mis-match during mixed-precision training with extract_patches_from_pyramid
#84Comparing
fork-3005-hit2sjtu/dev
(4058274
) withmain
(d645a84
)0%
- Untouched
- 4
Benchmarks
4 total
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.py0%
8.5 s8.5 s
test_aug_2d_solarize[1-1-128-128-cpu-float32-inductor]
benchmarks/augmentation/2d_intensity_test.py0%
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
Commits
Click on a commit to change the comparison rangeBase
main
d645a84
-0.1%
Fix type mis-match during mixed-precision training.
4058274
1 year ago
by hit2sjtu