Searched refs:NetWithLoss5 (Results 1 – 1 of 1) sorted by relevance
207 class NetWithLoss5(nn.Cell): class209 super(NetWithLoss5, self).__init__()250 net = GradWrap5(NetWithLoss5(Net()))