Hasan Shahriar

icon indicating a website link https://github.com/hsleonis

icon location Germany icon location Data Scientist | Machine Learning Engineer

Results 1 issues of Hasan Shahriar

Fix deprecated torch.cuda.amp.GradScaler usage in trainer

Replace deprecated `torch.cuda.amp.GradScaler` with the new `torch.amp.GradScaler` API to resolve FutureWarning in PyTorch 2.4+. **Changes:** Updated `trainer.py` line 545 to use `torch.amp.GradScaler('cuda', ...)` instead of `torch.cuda.amp.GradScaler(...)` **Issue:** Fixes deprecation warning:...