[FutureWarning] Fixing warning triggered by torch.cuda.reset_max_memory_allocated()
usage.
#21453
Loading
torch.cuda.reset_max_memory_allocated()
usage.
#21453