Invalid argument 2 error

[SejeongPark8354]

Hi, I started to implement your arae_train.py (diff_vae_gan) code with my dataset(from ZINC).
But, I got some errors like below.
Is there solution for these errors?
Thank you!! :)

KL: 0.12, Word: 40.61, Topo: 78.37, Assm: 78.59, Disc: 2.2284, Gen: 2.3088, GP: 0.9473, PNorm: 100.4350.00, 3.98
invalid argument 2: size '[-1 x 15 x 300]' is invalid for input with 36300 elements at /pytorch/aten/.cpp:80
invalid argument 2: size '[-1 x 15 x 300]' is invalid for input with 18300 elements at /pytorch/aten/.cpp:80
KL: 0.16, Word: 44.39, Topo: 83.93, Assm: 82.62, Disc: 2.4099, Gen: 2.4386, GP: 1.0063, PNorm: 101.387.92, 5.55
invalid argument 2: size '[-1 x 15 x 300]' is invalid for input with 27300 elements at /pytorch/aten/.cpp:80
invalid argument 2: size '[-1 x 15 x 300]' is invalid for input with 31800 elements at /pytorch/aten/.cpp:80
invalid argument 2: size '[-1 x 15 x 300]' is invalid for input with 22800 elements at /pytorch/aten/.cpp:80
invalid argument 2: size '[-1 x 15 x 300]' is invalid for input with 27300 elements at /pytorch/aten/.cpp:80
invalid argument 2: size '[-1 x 15 x 300]' is invalid for input with 18300 elements at /pytorch/aten/.cpp:80
KL: 0.34, Word: 89.83, Topo: 164.00, Assm: 166.76, Disc: 4.7831, Gen: 4.9966, GP: 1.9535, PNorm: 102.: 10.16, 4.44
invalid argument 2: size '[-1 x 15 x 300]' is invalid for input with 13800 elements at /pytorch/aten/.cpp:80
invalid argument 2: size '[-1 x 15 x 300]' is invalid for input with 45300 elements at /pytorch/aten/.cpp:80
invalid argument 2: size '[-1 x 15 x 300]' is invalid for input with 40800 elements at /pytorch/aten/.cpp:80
invalid argument 2: size '[-1 x 15 x 300]' is invalid for input with 22800 elements at /pytorch/aten/.cpp:80
invalid argument 2: size '[-1 x 15 x 300]' is invalid for input with 22800 elements at /pytorch/aten/.cpp:80
invalid argument 2: size '[-1 x 15 x 300]' is invalid for input with 31800 elements at /pytorch/aten/.cpp:80
KL: 0.17, Word: 37.59, Topo: 66.64, Assm: 66.55, Disc: 1.9452, Gen: 1.9557, GP: 0.7845, PNorm: 103.706.61, 5.00
KL: 0.27, Word: 54.90, Topo: 95.13, Assm: 95.50, Disc: 2.8014, Gen: 3.0039, GP: 1.1223, PNorm: 104.7217.78, 6.21
invalid argument 2: size '[-1 x 15 x 300]' is invalid for input with 36300 elements at /pytorch/aten/.cpp:80
invalid argument 2: size '[-1 x 15 x 300]' is invalid for input with 40800 elements at /pytorch/aten/.cpp:80
invalid argument 2: size '[-1 x 15 x 300]' is invalid for input with 49800 elements at /pytorch/aten/.cpp:80
invalid argument 2: size '[-1 x 15 x 300]' is invalid for input with 22800 elements at /pytorch/aten/.cpp:80
invalid argument 2: size '[-1 x 15 x 300]' is invalid for input with 36300 elements at /pytorch/aten/.cpp:80
invalid argument 2: size '[-1 x 15 x 300]' is invalid for input with 27300 elements at /pytorch/aten/.cpp:80