Purpose of os.environ['CUDA_LAUNCH_BLOCKING'] = '1' in train.py

jishengpeng / WavTokenizer

SOTA discrete acoustic codec models with 40 tokens per second for audio language modeling

MIT License

830 stars 46 forks source link

Closed seastar105 closed 2 months ago

seastar105 commented 2 months ago

Is it okay to remove this line in train.py to speed up train?

jishengpeng commented 2 months ago

Is it okay to remove this line in train.py to speed up train?

it is okay