Fix FutureWarning due to annotations - #152

Open

bauwenst wants to merge 1 commit into

yzhangcs:mainfrom

Commits on Nov 9, 2024

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