Searched refs:NetWithLoss6 (Results 1 – 1 of 1) sorted by relevance
257 class NetWithLoss6(nn.Cell): class259 super(NetWithLoss6, self).__init__()298 net = GradWrap6(NetWithLoss6(Net()))