def test_force_cuda(self):
mnist_cnn.to("cuda")
results = maxloss(
mnist_cnn,
nn.CrossEntropyLoss(),
batch_X,
batch_y_cuda,
device="cuda",
)
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)
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())
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)
def test_maxloss_range(self):
results = maxloss(mnist_cnn, nn.CrossEntropyLoss(), batch_X, batch_y)
self.assertTrue(((results >= -1) & (results <= 1)).all())
def test_maxloss_shape(self):
results = maxloss(mnist_cnn, nn.CrossEntropyLoss(), batch_X, batch_y)
self.assertTupleEqual(results.shape, batch_X.shape)