I am not sure if it will solve the issue. But can you try changing the line 273 in trainer.py as below:
loss_d = self.mse(self.fx.squeeze(), self.real_label) + self.mse(self.fx_tilde.squeeze(), self.fake_label)