anse3832/MUSIQ

Patch Embedding Module gradient update

Abdurrahheem opened this issue · 0 comments

optimizer.zero_grad()

I was wandering why PEM - backbone does not get updated, although it's parameters are passed in to the optimizer