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