test_input = test_set.data.view(-1, 1, 28, 28).float()
test_targets = test_set.targets
######################################################################
test_input = test_set.data.view(-1, 1, 28, 28).float()
test_targets = test_set.targets
######################################################################