Running causes frozen(unresponsive device)

lucidrains / imagen-pytorch

Implementation of Imagen, Google's Text-to-Image Neural Network, in Pytorch

MIT License

8.11k stars 768 forks source link

Open troy600 opened 1 year ago

troy600 commented 1 year ago

When i run the script my cpu usage sky rocketed into 99% and my system is hanging

CAn anyone help me how to reduce the script thread?

TheFusion21 commented 1 year ago

torch.set_num_threads(int) could help