Python maxloss示例

示例#1

 def test_force_cuda(self):
     mnist_cnn.to("cuda")
     results = maxloss(
         mnist_cnn,
         nn.CrossEntropyLoss(),
         batch_X,
         batch_y_cuda,
         device="cuda",
     )

示例#2

 def test_cuda(self):
     mnist_cnn.to("cuda")
     results = maxloss(mnist_cnn, nn.CrossEntropyLoss(), batch_X_cuda,
                       batch_y_cuda)
     self.assertTrue(results.is_cuda)

示例#3

 def test_maxloss_range(self):
     mnist_cnn.to("cuda")
     results = maxloss(mnist_cnn, nn.CrossEntropyLoss(), batch_X_cuda,
                       batch_y_cuda)
     self.assertTrue(((results >= -1) & (results <= 1)).all())

示例#4

 def test_maxloss_shape(self):
     mnist_cnn.to("cuda")
     results = maxloss(mnist_cnn, nn.CrossEntropyLoss(), batch_X_cuda,
                       batch_y_cuda)
     self.assertTupleEqual(results.shape, batch_X.shape)

示例#5

 def test_maxloss_range(self):
     results = maxloss(mnist_cnn, nn.CrossEntropyLoss(), batch_X, batch_y)
     self.assertTrue(((results >= -1) & (results <= 1)).all())

示例#6

 def test_maxloss_shape(self):
     results = maxloss(mnist_cnn, nn.CrossEntropyLoss(), batch_X, batch_y)
     self.assertTupleEqual(results.shape, batch_X.shape)