Open kia9900 opened 1 year ago
I think your vram is just too small, try running it on google colab
everything is fine on google colab, except !python train.py -c configs/config.json -m 32k
Traceback (most recent call last):
File "train.py", line 288, in <module>
main()
File "train.py", line 41, in main
assert torch.cuda.is_available(), "CPU training is not allowed."
AssertionError: CPU training is not allowed.
I use miniconda and I try
python train.py -c configs/config.json -m 32k
then i got this errori set
batch_size
to 1 andnum_workers
to 3 but didn't fix it