DeepTrackAI / DeepTrack2

DeepTrack2
MIT License
162 stars 50 forks source link

Change loss input order of CGAN and PCGAN #157

Closed GideonJagen closed 1 year ago

GideonJagen commented 1 year ago

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__()