Skip to content

Conversation

pweigel
Copy link
Collaborator

@pweigel pweigel commented Jan 22, 2025

For cases where the final layer of a model already reduces the dimension to the number of outputs, you may not want the affine layer in the LearnedTask. Added a simple bool for disabling the layer.

RasmusOrsoe
RasmusOrsoe previously approved these changes Feb 7, 2025
Copy link
Collaborator

@RasmusOrsoe RasmusOrsoe left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

@pweigel thank you for this contribution!

@RasmusOrsoe RasmusOrsoe self-requested a review September 2, 2025 07:58
@RasmusOrsoe RasmusOrsoe dismissed their stale review September 2, 2025 07:59

Need to re-run checks before merging.

@Aske-Rosted
Copy link
Collaborator

With disable_affine=True the IdentityTask will be able to run as discussed in #825

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants