Return a default multi-GPU trainer, if you don't care about the details. It may not be the most efficient one for your task. Args: gpus (list[int]): list of GPU ids.
(gpus)
| 106 | |
| 107 | |
| 108 | def SyncMultiGPUTrainer(gpus): |
| 109 | """ |
| 110 | Return a default multi-GPU trainer, if you don't care about the details. |
| 111 | It may not be the most efficient one for your task. |
| 112 | |
| 113 | Args: |
| 114 | gpus (list[int]): list of GPU ids. |
| 115 | """ |
| 116 | return SyncMultiGPUTrainerParameterServer(gpus, ps_device='cpu') |
| 117 | |
| 118 | |
| 119 | class AsyncMultiGPUTrainer(SingleCostTrainer): |
no test coverage detected