Open slives-lab opened 1 year ago
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?
hello world
jiegenyy
commented
1 year ago
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?