Skip to content
Discussion options

You must be logged in to vote

The incorrect order of layers in the forward pass of CircleModelV0 or CircleModelV1 causes this error.

Incorrect:

def forward(self, x):
        return self.layer_1(self.layer_2(self.layer_3(x)))

Correct:

def forward(self, x):
        return self.layer_3(self.layer_2(self.layer_1(x)))

Replies: 3 comments 1 reply

Comment options

You must be logged in to vote
0 replies
Comment options

You must be logged in to vote
0 replies
Comment options

You must be logged in to vote
1 reply
@mrdbourke
Comment options

Answer selected by mrdbourke
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Category
Q&A
Labels
None yet
4 participants