This is the error details:
RuntimeError                              Traceback (most recent call last)
<ipython-input-41-b5e01faa9c48> in <module>
     13         fake_imgs = G(z)
     14 
---> 15         real_validity = D(real_imgs)
     16         fake_validity = D(fake_imgs.detach())
     17 

~\anaconda3\envs\st\lib\site-packages\torch\nn\modules\module.py in _call_impl(self, *input, **kwargs)
    887             result = self._slow_forward(*input, **kwargs)
    888         else:
--> 889             result = self.forward(*input, **kwargs)
    890         for hook in itertools.chain(
    891                 _global_forward_hooks.values(),

<ipython-input-9-ed9aa1621bc7> in forward(self, input)
     42 
     43     def forward(self, input):
---> 44         output = self.extract_feature(input)
     45         output = self.ln1(output)
     46         output = output.view(-1)

<ipython-input-9-ed9aa1621bc7> in extract_feature(self, input)
     32     def extract_feature(self, input):
     33         output = input.contiguous()
---> 34         output = output.view(-1, 3, DIM, DIM)
     35         output = self.conv1(output)
     36         output = self.rb1(output)

RuntimeError: shape '[-1, 3, 64, 64]' is invalid for input of size 120000