In the code (MAT.py,274 line):
'''
if not jump_aux:
aux_loss,feature_matrix_d=self.auxiliary_loss(feature_maps_d,attentionmaps,y)
'''
it first compute the AttentionPooling between the feature_maps_d and attentions,then get feature_matrix and loss; but I think it should use the parameter feature_maps (not feature_maps_d in code)?
change it to be:
'''
if not jump_aux:
aux_loss,feature_matrix_d=self.auxiliary_loss(feature_maps,attentionmaps,y)
'''
I want to known wether I make a wrong understanding about the code?
In the code (MAT.py,274 line): ''' if not jump_aux: aux_loss,feature_matrix_d=self.auxiliary_loss(feature_maps_d,attentionmaps,y) ''' it first compute the AttentionPooling between the feature_maps_d and attentions,then get feature_matrix and loss; but I think it should use the parameter feature_maps (not feature_maps_d in code)? change it to be: ''' if not jump_aux: aux_loss,feature_matrix_d=self.auxiliary_loss(feature_maps,attentionmaps,y) ''' I want to known wether I make a wrong understanding about the code?