Receiving incomaptible shapes error when calling train_model(). Unit test on model architecture was passed with no errors before.
Mean() layer is created along axis = 1
Dense Layer has outputs 2
Is there a way to directly inspect where in tl.Serial() the shape mismatch is happening, i.e. which layers are causing trouble?
Also, shapes (8, 2), (16, 2) seem to indicate some issue related batch_size (=16) but we are not changing values for batch_size in the course of the assignment.
Any help is greatly appreciated!
Cheers
Ben