can't training model

[Pi-777]

when I use python train.py xxxxx
I got this report:

Setting up PyTorch plugin "upfirdn2d_plugin"... Done.
Traceback (most recent call last):

xxxxxxxxxxxxx/training/networks_stylegan2.py", line 657, in forward
y = x.reshape(G, -1, F, c, H, W) # [GnFcHW] Split minibatch N into n groups of size G, and channels C into F groups of size c.
RuntimeError: shape '[4, -1, 1, 512, 4, 4]' is invalid for input of size 81920

did anyone meet this report like above and fix it?

[jasuriy]

@Pi-777 hi
did you fix the issue?
were you able to train the model with your own dataset?