RuntimeError: The size of tensor a (1024) must match the size of tensor b (8192) at non-singleton dimension 1

[slives-lab]

When trying to train priors I receive the error, "RuntimeError: The size of tensor a (1024) must match the size of tensor b (8192) at non-singleton dimension 1"

python jukebox/train.py --hps=small_vqvae,small_prior,all_fp16,cpu_ema --name=small_prior
--sample_length=262144 --bs=4 --audio_files_dir=path--labels=False --train --test --aug_shift --aug_blend
--restore_vqvae=logs/small_vqvae/checkpoint_latest.pth.tar --prior --levels=2 --level=1 --weight_decay=0.01 --save_iters=1000

Any suggestions to fix?

[jiegenyy]

hello world