Skip to content

Fix: Issue 259 - Modify condition of naming for last layer#260

Merged
TjarkMiener merged 1 commit into
ctlearn-project:mainfrom
hugo-varenne:predict_simplecnn_error
Oct 17, 2025
Merged

Fix: Issue 259 - Modify condition of naming for last layer#260
TjarkMiener merged 1 commit into
ctlearn-project:mainfrom
hugo-varenne:predict_simplecnn_error

Conversation

@hugo-varenne
Copy link
Copy Markdown
Collaborator

Solution to the prediction of classification model (particle type) error (#259). This occurs when using LoadedModel or SimpleCNN subclass of CTLearnModel. This is due to a naming behaviour where the last layer name was expecting to finish with "_softmax". With LoadedModel and SimpleCNN, the naming wasn't working as expected, so in order to be more flexible, the condition changed from looking at the last layer name to looking at the type of layer (softmax for classification task)

@TjarkMiener TjarkMiener self-requested a review October 17, 2025 12:32
Copy link
Copy Markdown
Member

@TjarkMiener TjarkMiener left a comment

Choose a reason for hiding this comment

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

Thanks for the bug fix @hugo-varenne!

@TjarkMiener TjarkMiener merged commit e38f9f1 into ctlearn-project:main Oct 17, 2025
5 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants