Searched refs:TrainStepWrap2 (Results 1 – 1 of 1) sorted by relevance
62 class TrainStepWrap2(nn.Cell): class68 super(TrainStepWrap2, self).__init__()84 return TrainStepWrap2(network, sens)