Memory Crash

[usmanali414]

Hey,
I am trying to train the model on noise data that you provided and just running your code.But your code crashes as soon as session initializes the graph variables. The specs are as follows:
Google colab
ram: 25 gb
python: 3.10
cpu
tensorflow:2.12

[laculp]

Hi.

Thanks so much for contacting us!
I just tried myself again and I also found a crash. However, I waited, and then a restarted my runtime and ran everything again, and it works. I was loading the model into the colab folder, and then running immediately, and I think that the model is not actually properly loaded (into the colab folder) for quite some time when you do this. Some time, and then a restart of the runtime (I don't know whether this is necessary...) worked for me.
I am adding a check to see if the variables are actually getting loaded (the variable file is the largest file, and thus the last to be loaded properly I think) right now.
Can you confirm that this is what your problem was? If not, I'll definitely take another look and see if I can figure out what is going wrong.

Thanks for reaching out,
Laura