Searched refs:LargeBatchModel (Results 1 – 1 of 1) sorted by relevance
1503 class LargeBatchModel(torch.nn.Module): class in TestFxDetectOutliers1554 self.LargeBatchModel(param_size=128), detector_set1594 model = self.LargeBatchModel(param_size=param_size)1644 model = self.LargeBatchModel(param_size=param_size)1694 model = self.LargeBatchModel(param_size=param_size)