code finishes on colab unexpectedly

justinpinkney / stable-diffusion

MIT License

1.45k stars 266 forks source link

code finishes on colab unexpectedly #20

Closed zaidalyafeai closed 2 years ago

zaidalyafeai commented 2 years ago

I was trying to run the code on colab pro, I had to change the code to set ngpu = 1 but after that when I run main.py it runs until Setting up LambdaLR scheduler... then it just finishes, any idea what is going on?

justinpinkney commented 2 years ago

Probably just runs out of memory, if you see ^C that's the reason.