Hello, I’m new to pytorch/ML.I’m trying to port the CenterLoss 23 to torch, the networ architecture is here 27, roughly like:
convs ... - fc1 - fc2 - softmax_loss
             |
             |  - custom_loss(center_loss)


My question is:  how can I implement the multiple loss function at different layer in pytorch?
Thanks.