diff --git a/assignment-2/handout-1/tester_demo.py b/assignment-2/handout-1/tester_demo.py index 563de6b306d6a967c9a2a189c917cb858e7be4fe..e7ae741074bf8923a04cbb24fe2f9abbbd928df7 100644 --- a/assignment-2/handout-1/tester_demo.py +++ b/assignment-2/handout-1/tester_demo.py @@ -6,7 +6,7 @@ from numpy_fnn import Matmul, Relu, Softmax, Log, NumpyModel from torch_mnist import TorchModel from utils import get_torch_initialization, one_hot -err_epsilon = 1e-3 +err_epsilon = 1e-5 #modified err_p = 0.4 @@ -140,7 +140,7 @@ def test_model(): try: - numpy_model.backward(y2) + numpy_model.backward(-10*y2) #modified check_flag_2 = [ check_result(numpy_model.log_grad, torch_model.log.grad),