Hi,
I have a problem with the Decoder Layer, I got the Wrong values in out and I don’t know why…
---------------------------------------------------------------------------
AssertionError Traceback (most recent call last)
<ipython-input-205-7c54c129d8b0> in <module>
1 # UNIT TEST
----> 2 DecoderLayer_test(DecoderLayer, create_look_ahead_mask)
~/work/W4A1/public_tests.py in DecoderLayer_test(target, create_look_ahead_mask)
180 assert np.allclose(attn_w_b1[0, 0, 1], [0.5271505, 0.47284946, 0.], atol=1e-2), "Wrong values in attn_w_b1. Check the call to self.mha1"
181 assert np.allclose(attn_w_b2[0, 0, 1], [0.32048798, 0.390301, 0.28921106]), "Wrong values in attn_w_b2. Check the call to self.mha2"
--> 182 assert np.allclose(out[0, 0], [-0.22109576, -1.5455486, 0.852692, 0.9139523]), "Wrong values in out"
183
184
AssertionError: Wrong values in out
I checked the other topics but it doesn’t seem that I have the same problems. I checked that I am using the correct layer names, the correct order for Q,K and V (cf Course 5: Week 4 Transformer DecoderLayers - #14 by TMosh. So right now I really don’t know what to do to solve the unit test. Have any idea ?
Thanks