Searched defs:swa_device (Results 1 – 1 of 1) sorted by relevance
59 def _test_averaged_model(self, net_device, swa_device, ema): argument81 def _run_averaged_steps(self, dnn, swa_device, ema): argument