I’m running the test output cell for UNQ_C5 in this week’s assignment, and I note that my solution outputs
Serial_in3[
Branch_in2_out3[
[Dense_512, AttnHeads_in2]
[Dense_512, AttnHeads_in2]
[Dense_512, AttnHeads_in2]
]
DotProductAttn_in3
AttnOutput_in2
Dense_512
]
where the following is expected:
Serial[
Branch_out3[
[Dense_512, AttnHeads]
[Dense_512, AttnHeads]
[Dense_512, AttnHeads]
]
DotProductAttn_in3
AttnOutput
Dense_512
]
Relatedly, my code passes 6 and fails 2 unit tests, with mostly unhelpful text output. Presumably, there is something going on incorrectly with my code, and I am trying to diagnose what it is. Is there a reason why Trax would output _in# and _out#? This seems like it may be a red-herring, but I don’t know what else to do, given that I think I’m using proper syntax. I could post code snippets, but I worry about violating rules around sharing solutions. Thanks for any insight you can offer!