Searched refs:_test_averaged_model (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/test/optim/ |
| H A D | test_swa_utils.py | 59 def _test_averaged_model(self, net_device, swa_device, ema): member in TestSWAUtils
113 self._test_averaged_model(cpu, cpu, ema)
116 self._test_averaged_model(cuda, cpu, ema)
117 self._test_averaged_model(cpu, cuda, ema)
118 self._test_averaged_model(cuda, cuda, ema)
|