(self)
| 263 | model.device = torch.device(f"cuda:{num_gpus - 1 - i // models_each_gpu}") |
| 264 | |
| 265 | def init(self): |
| 266 | # set the model parameters requires_grad is False |
| 267 | for model in self.models: |
| 268 | model.requires_grad_(False) |
| 269 | model.eval() |
| 270 | |
| 271 | def to(self, device: torch.device): |
| 272 | for model in self.models: |
no outgoing calls
no test coverage detected