Fix FutureWarning due to annotations#152

Open

bauwenst wants to merge 1 commit intoyzhangcs:mainfrom

Commits on Nov 9, 2024

Fixed FutureWarnings by converting torch.cuda.amp annotations into torch.amp.
bauwenst
committed