search

Audio-WestlakeU / FN-SSL

The Official PyTorch Implementation of FN-SSL & IPDnet for Sound Source Localization
81 stars 8 forks source link

Loss function definition #6

Open ahmadikalkhorani opened 2 weeks ago

ahmadikalkhorani commented 2 weeks ago

Hello,

Could you please provide the loss function definition? (self.loss in Learner.py is not defined).

Thanks

wangyabo123 commented 2 weeks ago

Hi, In the torch version of the code, the SourceTrackingFromSTFTLearner class inherits from the Learner class and defines the loss function in SourceTrackingFromSTFTLearner. (Please see the line 470 of Learner. py)

ahmadikalkhorani commented 2 weeks ago

Thanks for the prompt response.

Does this line correspond to Section IV.E: Frame-level PIT and Sound Source Localization?

Thanks

wangyabo123 commented 2 weeks ago

Sorry for misunderstanding your meaning, the current open source code is FN-SSL, I will upload the code of IPDnet in the next few days and remind you.