fix: replace deprecated torch.cuda.amp with torch.amp - #1089

Open

ktz03 wants to merge 5 commits into

meituan:mainfrom

ktz03:fix/pytorch-amp-api

Commits on Aug 16, 2026

fix: migrate engine.py to torch.amp
ktz03
committed
327e48f
fix: migrate loss.py to torch.amp
ktz03
committed
279441b
fix: migrate loss_fuseab.py to torch.amp
ktz03
committed
895964c
fix: migrate loss_distill.py to torch.amp
ktz03
committed
1e67b4f
fix: replace deprecated torch.cuda.amp with torch.amp
ktz03
committed
40a8e06