Sorry I haven’t mentioned this before. The result of
input_gmm, self.log_likelihood_input = self.gmm.inference(input_style_reshaped)
is numpy array for both the input_gmm and self.log_likelihood.
Therefore, when I do the
F.mse_loss(self.input_gmm, self.target).to(device)
,the computation can’t be calculated because one of them is in numpy array type. How to counter that?