Change loss input order of CGAN and PCGAN by GideonJagen · Pull Request #157 · DeepTrackAI/DeepTrack2 · GitHub
Skip to content

Change loss input order of CGAN and PCGAN#157

Merged
BenjaminMidtvedt merged 1 commit into
DeepTrackAI:developfrom
GideonJagen:fix-cgan-loss
Nov 30, 2022
Merged

Change loss input order of CGAN and PCGAN#157
BenjaminMidtvedt merged 1 commit into
DeepTrackAI:developfrom
GideonJagen:fix-cgan-loss

Conversation

@GideonJagen

Copy link
Copy Markdown
Contributor

The inputs given when calling all loss_functions in both CGAN and PCGAN were written backwards, i.e. loss_function(y_pred, y_true) instead of the correct loss_function(y_true, y_pred). This lead to negative loss values when using BinaryCrossentropyLoss.

I also updated super(PCGAN).__init__() to super().__init__()

@BenjaminMidtvedt BenjaminMidtvedt merged commit f93cfed into DeepTrackAI:develop Nov 30, 2022
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.

2 participants