changed a bunch of places that use torch.cuda.empty_cache() to use torch_gc() instead |
||
|---|---|---|
| .. | ||
| scripts | ||
| preload.py | ||
| scunet_model_arch.py | ||
changed a bunch of places that use torch.cuda.empty_cache() to use torch_gc() instead |
||
|---|---|---|
| .. | ||
| scripts | ||
| preload.py | ||
| scunet_model_arch.py | ||