Open SevenTae opened 2 years ago
I used unet to train the voc data set, loaded my own weights and took the activation of a certain layer in the middle, but got this error “ grad can be implicitly created only for scalar outputs”
Excuse me, have you solved this problem? I had the same problem.
I used unet to train the voc data set, loaded my own weights and took the activation of a certain layer in the middle, but got this error “ grad can be implicitly created only for scalar outputs”