 Matthew McGoogan
|
c67c40f983
|
torch.cuda.empty_cache() defaults to cuda:0 device unless explicitly set otherwise first. Updating torch_gc() to use the device set by --device-id if specified to avoid OOM edge cases on multi-GPU systems.
|
2022-11-26 23:25:16 +00:00 |
|
