added torch.mps.empty_cache() to torch_gc()
changed a bunch of places that use torch.cuda.empty_cache() to use torch_gc() instead
Showing
想要评论请 注册 或 登录
changed a bunch of places that use torch.cuda.empty_cache() to use torch_gc() instead