RuntimeError: The size of tensor a (32) must match the size of tensor b (4) at non-singleton dimension 0

[toniesteves]

on executing train.py I receive de following error:

traceback (most recent call last):
  File "train.py", line 839, in <module>
    trainer.train()
  File "train.py", line 614, in train
    interpolates = alpha*self.x.data+((1-alpha)*self.z_x_gen.data)
RuntimeError: The size of tensor a (32) must match the size of tensor b (4) at non-singleton dimension 0

System Software Overview:
System Version: macOS 10.14.6 (18G95)
Kernel Version: Darwin 18.7.0
Boot Volume: Mojave
Boot Mode: Normal
Secure Virtual Memory: Enabled
System Integrity Protection: Enabled