-
Notifications
You must be signed in to change notification settings - Fork 709
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
🚜 Refactor loss computation #364
Conversation
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Thanks, a lot cleaner now! Only a single comment
anomalib/models/fastflow/loss.py
Outdated
jacobians (List[Tensor]): Log of the jacobian determinants from the fastflow model. | ||
|
||
Returns: | ||
Tensor: _description_ |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
This is a leftover from the previous PR, but would be good to add a description here in this PR.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Thanks for the refactor!
Description
Refactor the loss computation, see Move loss to lightning module/separate module #351 for more details
Fixes Move loss to lightning module/separate module #351
Changes
Checklist