(*args)
| 114 | |
| 115 | |
| 116 | def unload_complete_models(*args): |
| 117 | for m in gpu_complete_modules + list(args): |
| 118 | m.to(device=cpu) |
| 119 | print(f'Unloaded {m.__class__.__name__} as complete.') |
| 120 | |
| 121 | gpu_complete_modules.clear() |
| 122 | torch.cuda.empty_cache() |
| 123 | return |
| 124 | |
| 125 | |
| 126 | def load_model_as_complete(model, target_device, unload=True): |
no outgoing calls
no test coverage detected